
    #j                     V    d dl mZ d dlmZ  e            rddlZ G d de          ZdS )   )ConversionOps)is_torch_available    Nc            	       V    e Zd Zd Z	 ddej        dedz  deeej        f         fdZdS )QuarkDeserializec                     || _         d S N)hf_quantizer)selfr
   s     i/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/transformers/integrations/quark.py__init__zQuarkDeserialize.__init__   s    (    N
input_dictfull_layer_namereturnc                     t          |                                          d         }t          |t                     r|d         n|}||iS )Nr   )listvalues
isinstance)r   r   r   kwargsvalues        r   convertzQuarkDeserialize.convert   sI     Z&&(())!,&ud33>a''r   r	   )	__name__
__module____qualname__r   torchTensorstrdictr    r   r   r   r      so        ) ) ) '+( (L( t(
 
c5<	 ( ( ( ( ( (r   r   )core_model_loadingr   utilsr   r   r   r    r   r   <module>r#      s{     / . . . . . & & & & & &  LLL( ( ( ( (} ( ( ( ( (r   