Fix/nemotron mtp quantaized (#19433)
This commit is contained in:
@@ -665,7 +665,10 @@ class ModelConfig:
|
||||
quant_cfg = quant_cfg.to_dict()
|
||||
if quant_cfg is not None:
|
||||
# Identify modelopt quantization
|
||||
if "quant_method" not in quant_cfg:
|
||||
if (
|
||||
"quant_method" not in quant_cfg
|
||||
or quant_cfg["quant_method"] == "modelopt"
|
||||
):
|
||||
parsed_cfg = self._parse_modelopt_quant_config(
|
||||
{"quantization": quant_cfg}
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user