ó
    ñMói  ã                   ó>   • S SK Jr  S SKJr  S SKJr   " S S\5      rg)é    )ÚOptional)ÚPreTrainedModel)Úis_flash_attn_2_availablec                   ó(   • \ rS rSrS\\   4S jrSrg)ÚSuryaPreTrainedModelé   Úattn_implementationc                 ó¨   • Uc6   U R                  S5        SnU R                  (       a  [	        5       (       a  SnU$ ! [        [        4 a    Sn N8f = f)NTÚsdpaÚeagerÚflash_attention_2)Ú_sdpa_can_dispatchÚ
ValueErrorÚImportErrorÚ_supports_flash_attnr   )Úselfr	   Úkwargss      Úq/var/www/eduai.edurigo.com/storigo/production/storigo_env/lib/python3.13/site-packages/surya/common/pretrained.pyÚ%_check_and_adjust_attn_implementationÚ:SuryaPreTrainedModel._check_and_adjust_attn_implementation	   s_   € ð Ñ&ð.Ø×'Ñ'¨Ô-Ø&,Ð#ð ×(×(Ô-F×-HÑ-HØ&9Ð#à"Ð"øô ¤Ð,ó .Ø&-Ò#ð.ús   …< ¼AÁA© N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   Ústrr   Ú__static_attributes__r   ó    r   r   r      s   † ð#Ø#+¨C¡=÷#r   r   N)Útypingr   Útransformersr   Útransformers.utilsr   r   r   r   r   Ú<module>r"      s   ðÝ å (Ý 8ô#˜?õ #r   