diff --git a/src/axolotl/utils/schemas/quant.py b/src/axolotl/utils/schemas/quant.py index 62c33d99b..1a0139579 100644 --- a/src/axolotl/utils/schemas/quant.py +++ b/src/axolotl/utils/schemas/quant.py @@ -26,6 +26,8 @@ class HQQConfig(BaseModel): "If using HQQ, must set both `hqq_nbits` and `hqq_group_size`" ) + return data + def get_hqq_quant_config_kwargs(cfg):