From a64601f95782581c7e93d5291607acdb2b787f34 Mon Sep 17 00:00:00 2001 From: Sunny Liu Date: Wed, 16 Apr 2025 17:50:42 -0400 Subject: [PATCH] fix wrong variable name --- src/axolotl/utils/schemas/quant.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,