do not export config.key is value is None (#1187)

ModelCloud · Jan 31, 2025 · a4cb9bc · a4cb9bc
1 parent 9c59a1c
commit a4cb9bc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gptqmodel/quantization/config.py b/gptqmodel/quantization/config.py
@@ -395,6 +395,10 @@ def to_dict(self):
             PACK_DTYPE_FIELD: str(self.pack_dtype).split(".")[-1],
             META_FIELD: self.meta,
         }
+
+        # simplify: clean keys where the value is None
+        out = {k: v for k, v in out.items() if v is not None}
+
         dict_scale_dtype_to_str(out)
         return out