added task_type and updated init

2025-10-05 18:58:42 +02:00
parent 0f243eaac2
commit d48815cca2
2 changed files with 8 additions and 1 deletions
--- 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"]
+__all__ = ["get_attention_mask", "TaskType"]