Ë
    îÍ:j  ã                   óR   — 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y)	zFP-Quant integration fileé   )Úis_fp_quant_availableé    )ÚFPQuantConfig)ÚFPQuantDtypeÚconfigc           
      óÀ  — | j                   dk(  rt        j                  }n8| 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      úw/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/transformers/integrations/fp_quant.pyÚadapt_fp_quant_configr      sÊ   € Ø×Ñ˜wÒ&Ü$×*Ñ*‰Ø	×	Ñ	 Ò	(Ü$×*Ñ*‰äÐ6°v×7KÑ7KÐ6LÐMÓNÐNà×Ñ Ò&Ü%×*Ñ*‰äÐ7¸×8MÑ8MÐ7NÐOÓPÐPäØ#Ø×,Ñ,Ø%Ø#×8Ñ8Ø"×6Ñ6Ø!×4Ñ4Ø×,Ñ,Ø%×<Ñ<ô	ð 	ó    N)	Ú__doc__Úutilsr   Úfp_quantr   r   r   Ú&transformers.utils.quantization_configr   © r   r   ú<module>r!      s,   ðñ õñ
 ÔÝ=Ý%å @ð -ô r   