o
    ~rËi•  ã                   @   s:   d dl mZ d dlmZ eƒ rddlZG dd„ deƒZdS )é   )ÚConversionOps)Úis_torch_availableé    Nc                	   @   s>   e Zd Zdd„ Z	d	dejdedB deeejf fdd„ZdS )
ÚQuarkDeserializec                 C   s
   || _ d S ©N)Úhf_quantizer)Úselfr   © r	   úm/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/transformers/integrations/quark.pyÚ__init__   s   
zQuarkDeserialize.__init__NÚ
input_dictÚfull_layer_nameÚreturnc                 K   s.   t | ¡ ƒd }t|t ƒr|d n|}||iS )Nr   )ÚlistÚvaluesÚ
isinstance)r   r   r   ÚkwargsÚvaluer	   r	   r
   Úconvert   s   zQuarkDeserialize.convertr   )	Ú__name__Ú
__module__Ú__qualname__r   ÚtorchÚTensorÚstrÚdictr   r	   r	   r	   r
   r      s    ýþýûr   )Úcore_model_loadingr   Úutilsr   r   r   r	   r	   r	   r
   Ú<module>   s
   