diff --git a/Project_Model/Libs/Transformer/Utils/__init__.py b/Project_Model/Libs/Transformer/Utils/__init__.py
index 856b51f..2831ec4 100644
--- a/Project_Model/Libs/Transformer/Utils/__init__.py
+++ b/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"]
\ No newline at end of file
+__all__ = ["get_attention_mask", "TaskType"]
\ No newline at end of file
diff --git a/Project_Model/Libs/Transformer/Utils/task_type.py b/Project_Model/Libs/Transformer/Utils/task_type.py
new file mode 100644
index 0000000..46a42eb
--- /dev/null
+++ b/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()
\ No newline at end of file