diff --git a/src/axolotl/utils/schemas/quant.py b/src/axolotl/utils/schemas/quant.py index 1a0139579..9297c1640 100644 --- a/src/axolotl/utils/schemas/quant.py +++ b/src/axolotl/utils/schemas/quant.py @@ -32,7 +32,7 @@ class HQQConfig(BaseModel): def get_hqq_quant_config_kwargs(cfg): # If no target module is specified, then target the whole model - if cfg.hqq_module_name is None: + if cfg.hqq_target_module is None: return { "nbits": cfg.hqq_nbits, "group_size": cfg.hqq_group_size,