
    i                     H    d dl mZ d dlmZ  e       rddlZ G d de      Zy)   )ConversionOps)is_torch_available    Nc            	       b    e Zd Zd Z	 ddej
                  dedz  deeej
                  f   fdZy)QuarkDeserializec                     || _         y N)hf_quantizer)selfr
   s     p/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/integrations/quark.py__init__zQuarkDeserialize.__init__   s
    (    N
input_dictfull_layer_namereturnc                 p    t        |j                               d   }t        |t               r|d   n|}||iS )Nr   )listvalues
isinstance)r   r   r   kwargsvalues        r   convertzQuarkDeserialize.convert   s;     Z&&()!,&ud3a''r   r	   )	__name__
__module____qualname__r   torchTensorstrdictr    r   r   r   r      sC    ) '+(LL( t(
 
c5<<	 (r   r   )core_model_loadingr   utilsr   r   r   r    r   r   <module>r#      s$     / & (} (r   