V0.0.1 Athene

Project_Model/Libs/Transformer/Utils/__init__.py

@@ -4,6 +4,7 @@ from .post_tokenization import truncate_sequence, pad_sequence, normalize_sequen
 from .inference_masking import inference_masking
 from .truncate_rdf_list import truncate_rdf_list
 from .decode_out import tensor2token
+from .decoder_input import get_decoder_input
 
 
 __all__ = [
@@ -17,4 +18,5 @@ __all__ = [
     "inference_masking",
     "truncate_rdf_list",
     "tensor2token",
+    "get_decoder_input"
 ]