tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix wrong variable name

Browse Source
This commit is contained in:
Sunny Liu
2025-04-16 17:50:42 -04:00
committed by Sung Ching Liu
parent eb7bc70b99
commit a64601f957
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/schemas/quant.py
View File

@@ -32,7 +32,7 @@ class HQQConfig(BaseModel):
def get_hqq_quant_config_kwargs(cfg): def get_hqq_quant_config_kwargs(cfg):
# If no target module is specified, then target the whole model # 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 { return {
"nbits": cfg.hqq_nbits, "nbits": cfg.hqq_nbits,
"group_size": cfg.hqq_group_size, "group_size": cfg.hqq_group_size,