[TU]"U40UD6 URUlgN)super__init__json_export_config)selfquantization_configkwargs __class__s _/var/www/html/shao/venv/lib/python3.13/site-packages/transformers/quantizers/quantizer_quark.pyrQuarkHfQuantizer.__init__>s$ ,77"5"H"Hc8[5(d [S5eg)NzLoading a Quark quantized model requires the `quark` library but it was not found in the environment. Please refer to https://quark.docs.amd.com/latest/install.html.)r ImportError)rargsr!s r#validate_environment%QuarkHfQuantizer.validate_environmentCs#!##x $r%modelr c SSKJn U"UURRURR URR S9 U$)Nr) _map_to_quark) pack_method custom_mode)quark.torch.export.apir-r quant_configrr.r/)rr+r!r-s r#$_process_model_before_weight_loading5QuarkHfQuantizer._process_model_before_weight_loadingIsD8   $ $ 1 1//;;00<<   r% param_valuez torch.Tensor param_name state_dictreturnc g)NT)rr+r4r5r6r!s r#check_quantized_param&QuarkHfQuantizer.check_quantized_paramUsr%cURS5SnU[;aURU[U5n[XXBS9 g)N.)value)splitCHECKPOINT_KEYSreplacer )rr+paramr5 param_devicer6unexpected_keyspostfixs r#create_quantized_param'QuarkHfQuantizer.create_quantized_param_sB""3'+ o %#++G_W5MNJ#E|Qr%c U$rr9)rr+r!s r##_process_model_after_weight_loading4QuarkHfQuantizer._process_model_after_weight_loadingis r%cgNFr9)rsafe_serializations r#is_serializable QuarkHfQuantizer.is_serializablelsr%cgrMr9)rs r# is_trainableQuarkHfQuantizer.is_trainableosr%)r)r+r )r7ztorch.nn.Parameterr)__name__ __module__ __qualname____firstlineno____doc__requires_calibrationrequired_packages requires_parameters_quantizationrr)r2strdictrboolr:rGrJrOpropertyrR__static_attributes__ __classcell__)r"s@r#rr1s   (,$I   $  cN  R Rr%r)typingrr file_utilsrbasermodeling_utilsr torchutilsr r r accelerate.utilsr get_loggerrTloggerrArr9r%r#rksw &+0HH<   H %-(*,6246 @{@r%