suppress pydantic warnings for model_ fields (#2344) (#2345)

neuralmagic · Jun 20, 2024 · c0cc0d3 · c0cc0d3
1 parent 7a41462
commit c0cc0d3
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(