a h@sLdZddlmZer.ddlmZddlmZddlmZedddZd S) zFP-Quant integration file)is_fp_quant_available) FPQuantConfig) FPQuantDtype)configc Csd|jdkrtj}ntd|j|jdkr4tj}ntd|jt||j||j|j |j |j dS)NZmxfp4zUnsupported forward dtype: Zbf16zUnsupported backward dtype: ) forward_dtypeforward_methodbackward_dtypestore_master_weightshadamard_group_sizepseudoquantizationmodules_to_not_convert) rrZMXFP4 ValueErrorr ZBF16FPQuantLinearConfigrr r r r )rrr r^/var/www/html/assistant/venv/lib/python3.9/site-packages/transformers/integrations/fp_quant.pyadapt_fp_quant_configs  rN) __doc__utilsrZfp_quantrrrZ&transformers.utils.quantization_configrrrrrs