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

bfineran · web-flow · commit c0cc0d3bf3c3 · 2024-06-20T15:52:50.000-04:00
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(

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ class KeyValueCacheConfig(BaseModel):`
`84`	`84`	`"the kv cache. If this is not provided, no transpose will "`
`85`	`85`	`"be applied.",`
`86`	`86`	`)`
`87`		`- model_config = ConfigDict(arbitrary_types_allowed=True)`
	`87`	`+ model_config = ConfigDict(arbitrary_types_allowed=True, protected_namespaces=())`
`88`	`88`
`89`	`89`
`90`	`90`	`OPT_CONFIG = KeyValueCacheConfig(`