o
    vžÄié  ã                   @   s$   d Z ddlmZ G dd„ deƒZdS )zm
Imaginaire4 Attention Subpackage:
Unified implementation for all Attention implementations.

Mask utilities
é    )ÚEnumc                   @   s   e Zd ZdZdZdZdZdS )Ú
CausalTypezN
    Different types of causal masking supported by backends of interest.
    r   é   é   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ZTopLeftZBottomRightÚDontCare© r   r   úS/data/cameron/vidgen/cosmos-policy/cosmos_policy/_src/imaginaire/attention/masks.pyr      s
    r   N)r	   Úenumr   r   r   r   r   r   Ú<module>   s   