Ë
    ÜÓ^j¥	  ã                   óh   — d dl Z d dlmZ d dlmZ dedeeef   fd„Zd
dee   dz  fd„Z	ed	„ «       Z
y)é    N)Úcontextmanager)ÚAnyÚtensor_nameÚreturnc                 ó^   — d|v r&|j                  dd«      \  }}| j                  |«      } | |fS )Nú.é   )ÚrsplitÚget_submodule)Úmoduler   Úmodule_names      ús/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/quantizers/quantizers_utils.pyÚget_module_from_namer      s<   € Ø
ˆkÑØ#.×#5Ñ#5°c¸1Ó#=Ñ ˆ�[Ø×%Ñ% kÓ2ˆØ�;ÐÐó    Úpatternsc                 ó8   ‡ — |€yt        ˆ fd„|D «       «      }| S )NTc              3   ó¨   •K  — | ]I  }t        j                  |› d �‰«      xs* t        j                  |› ‰«      xs ‰j                  |«      –— ŒK y­w)z\.N)ÚreÚmatchÚendswith)Ú.0ÚkeyÚ	full_names     €r   ú	<genexpr>z(should_convert_module.<locals>.<genexpr>&   sP   øè ø€ ò àô 	�‰�C�5˜�˜iÓ(Òd¬B¯H©H¸°u¸yÓ,IÒdÈY×M_ÑM_Ð`cÓMdÓdñùs   ƒAA)Úany)r   r   Úshould_not_converts   `  r   Úshould_convert_moduler      s1   ø€ ØÐØô ó àôó Ðð "Ð!Ð!r   c              #   óÜ  K  — ddl m}  |«       r¿ddl}t        | |j                  «      r| j
                  } n!t        | t        «      r|j                  | «      } t        | dd«      }|dk(  r*|j                  j                  | «      5  d–— 	 ddd«       y|dk(  r6t        |d«      r*|j                  j                  | «      5  d–— 	 ddd«       yd–— y# 1 sw Y   ŒIxY w# 1 sw Y   ŒxY w­w)zIAlign the current accelerator device with a tensor or device-like object.é   )Úis_torch_availabler   NÚtypeÚcudaÚxpu)Úutilsr    ÚtorchÚ
isinstanceÚTensorÚdeviceÚstrÚgetattrr"   Úhasattrr#   )r(   r    r%   Údevice_types       r   Ú	on_devicer-   -   sÔ   è ø€ õ +áÔÛä�f˜eŸl™lÔ+Ø—]‘]‰FÜ˜¤Ô$Ø—\‘\ &Ó)ˆFä˜f f¨dÓ3ˆØ˜&Ò Ø—‘×"Ñ" 6Ó*ñ ÛØ÷ð ð ˜%Ò¤G¨E°5Ô$9Ø—‘×!Ñ! &Ó)ñ ÛØ÷ð ô 
÷ð ú÷ð üs6   ‚BC,ÂCÂ5C,Ã C ÃC,ÃCÃC,Ã C)Ã%C,)N)r   Ú
contextlibr   Útypingr   r)   Útupler   Úlistr   r-   © r   r   ú<module>r3      sT   ðó 
Ý %Ý ð¨cð °e¸CÀ¸H±oó ñ"¨t°C©y¸4Ñ/?ó "ð& ñ
ó ñ
r   