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(