NanoSocrates/Project_Model/Libs/Transformer/Utils/__init__.py

from .attention_mask import get_causal_attention_mask,get_causal_attention_mask_batched
from .task_type import TaskType

__all__ = ["get_causal_attention_mask", "TaskType", "get_causal_attention_mask_batched"]
added method fot batched attention_mask 2025-10-06 12:00:11 +02:00			`from .attention_mask import get_causal_attention_mask,get_causal_attention_mask_batched`
added task_type and updated init 2025-10-05 18:58:42 +02:00			`from .task_type import TaskType`
Added attention_mask 2025-10-05 17:49:01 +02:00
added method fot batched attention_mask 2025-10-06 12:00:11 +02:00			`__all__ = ["get_causal_attention_mask", "TaskType", "get_causal_attention_mask_batched"]`