fix dumb mistake

2025-04-16 16:58:24 -04:00
parent db6c76b147
commit eb7bc70b99
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/models.py
+++ b/src/axolotl/utils/models.py
@@ -908,7 +908,7 @@ class ModelLoader:
            from axolotl.utils.schemas.quant import get_hqq_quant_config_kwargs

            self.model_kwargs["quantization_config"] = HqqConfig(
-                get_hqq_quant_config_kwargs(self.cfg)
+                **get_hqq_quant_config_kwargs(self.cfg)
            )

        # no longer needed per https://github.com/huggingface/transformers/pull/26610