added task_type and updated init

Project_Model/Libs/Transformer/Utils/__init__.py

@@ -1,3 +1,4 @@
 from .attention_mask import get_attention_mask
+from .task_type import TaskType
 
-__all__ = ["get_attention_mask"]
+__all__ = ["get_attention_mask", "TaskType"]

Project_Model/Libs/Transformer/Utils/task_type.py

@@ -0,0 +1,6 @@
+from enum import Enum, auto
+
+class TaskType(Enum):
+    RDF2TEXT = auto()
+    MASK = auto()
+    COMPLETATION = auto()