From c0cc0d3bf3c3f8b9682e1c0cb112f7353e2ec72e Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Thu, 20 Jun 2024 15:52:50 -0400 Subject: [PATCH] suppress pydantic warnings for model_ fields (#2344) (#2345) --- src/sparseml/exporters/transforms/kv_cache/configs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparseml/exporters/transforms/kv_cache/configs.py b/src/sparseml/exporters/transforms/kv_cache/configs.py index 686adf5c7d5..d0135a2c133 100644 --- 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(