if (not quant_algo or not quant_layers) and quant_config_file.is_file():
with open(quant_config_file, "r", encoding="utf-8") as f:
- quant_config = json.load(f).get("quantization") or {}
+ hf_quant_config = json.load(f)
+ quant_config = hf_quant_config.get("quantization") or {}
+ producer = hf_quant_config.get("producer") or {}
+ producer_name = (producer.get("name") or "").lower()
+ if quant_method is None:
+ self.hparams.setdefault("quantization_config", {})["quant_method"] = producer_name
quant_algo = quant_config.get("quant_algo", quant_algo)
quant_layers = quant_config.get("quantized_layers", quant_layers) or {}