Relax dependencies (#200)

Author: SimJeg

kvpress/presses/kvzap_press.py

@@ -12,11 +12,14 @@
 
 
 class KVzapConfig(PretrainedConfig):
-    model_type: str = "kvzap"
-    input_dim: int
-    output_dim: int
-    hidden_dim: Optional[int] = None
-    n_modules: int
+    model_type = "kvzap"
+
+    def __init__(self, *, input_dim: int, output_dim: int, n_modules: int, hidden_dim: Optional[int] = None, **kwargs):
+        super().__init__(**kwargs)
+        self.input_dim = input_dim
+        self.output_dim = output_dim
+        self.hidden_dim = hidden_dim
+        self.n_modules = n_modules
 
 
 class KVzapModel(PreTrainedModel):

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kvpress"
-version = "0.5.1"
+version = "0.5.2"
 description = "Efficiently compress the KV cache of any pretrained transformer"
 authors = [
     { name = "Simon Jegou" },
@@ -13,8 +13,8 @@ readme = "README.md"
 dependencies = [
     "numpy>=2.0.0,<3",
     "torch>=2.3.1,<3",
-    "transformers>=5.0.0",
-    "datasets>=2.21.0,<3",
+    "transformers>=4.56.0",
+    "datasets>=2.21.0",
     "pandas>=2.2.2,<3",
     "accelerate>=1.0.0,<2",
     "requests>=2.32.3,<3",