o
    ?ß±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TopLeftÚBottomRightÚDontCare© r   r   úY/data/cameron/vidgen/cosmos-predict2.5/cosmos_predict2/_src/imaginaire/attention/masks.pyr      s
    r   N)r	   Úenumr   r   r   r   r   r   Ú<module>   s   