suppress pydantic warnings for model_ fields

neuralmagic · Jun 20, 2024 · 05604a1 · 05604a1
1 parent 52245b3
commit 05604a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparseml/exporters/transforms/kv_cache/configs.py b/src/sparseml/exporters/transforms/kv_cache/configs.py
@@ -84,7 +84,7 @@ class KeyValueCacheConfig(BaseModel):
         "the kv cache. If this is not provided, no transpose will "
         "be applied.",
     )
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True, protected_namespaces=())
 
 
 OPT_CONFIG = KeyValueCacheConfig(