suppress pydantic warning in legacy quantization modifier (#2340) (#2341

)
neuralmagic · Jun 19, 2024 · 0f96284 · 0f96284
1 parent ffa3852
commit 0f96284
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: