Merge branch 'main' into sa/fix_example

neuralmagic · Jun 20, 2024 · 4f529d6 · 4f529d6
2 parents 6ec50d5 + 52245b3
commit 4f529d6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/sparseml/modifiers/quantization_legacy/base.py b/src/sparseml/modifiers/quantization_legacy/base.py
@@ -14,6 +14,8 @@
 
 from typing import Any, Dict, List, Optional
 
+from pydantic import ConfigDict
+
 from sparseml.core import Event, Modifier
 
 
@@ -81,6 +83,8 @@ class LegacyQuantizationModifier(Modifier):
     post_oneshot_calibration: Optional[bool] = False
     strict: bool = True
 
+    model_config = ConfigDict(protected_namespaces=())
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         if self.model_fuse_fn_kwargs is None: