
     `i                     Z    d Z ddlmZ  e            rddlmZ ddlmZ ddlmZ defdZdS )	zFP-Quant integration file   )is_fp_quant_available    )FPQuantConfig)FPQuantDtypeconfigc           
      X   | j         dk    rt          j        }n/| j         dk    rt          j        }nt	          d| j                    | j        dk    rt          j        }nt	          d| j                   t          || j        || j	        | j
        | j        | j        | j                  S )Nmxfp4nvfp4zUnsupported forward dtype: bf16zUnsupported backward dtype: )forward_dtypeforward_methodbackward_dtypestore_master_weightshadamard_group_sizepseudoquantizationtransform_initmodules_to_not_convert)r   r   MXFP4NVFP4
ValueErrorr   BF16FPQuantLinearConfigr   r   r   r   r   r   )r   r   r   s      v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/integrations/fp_quant.pyadapt_fp_quant_configr      s    w&&$*			(	($*Mv7KMMNNN&&%*O8MOOPPP#,%#8"6!4,%<	 	 	 	    N)	__doc__utilsr   fp_quantr   r   r   &transformers.utils.quantization_configr    r   r   <module>r!      s          
  &======%%%%%% @ @ @ @ @ @-      r   