Bump optimum version (#534)

* bump optimum * add peft
huggingface · Mar 28, 2024 · d06b07e · d06b07e
1 parent dd029a1
commit d06b07e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
 INSTALL_REQUIRES = [
     "transformers == 4.36.2",
     "accelerate == 0.23.0",
-    "optimum ~= 1.17.0",
+    "optimum ~= 1.18.0",
     "huggingface_hub >= 0.20.1",
     "numpy>=1.22.2, <=1.25.2",
     "protobuf<4",
@@ -32,6 +32,7 @@
     "diffusers >= 0.26.1",
     "safetensors",
     "sentence-transformers >= 2.2.0",
+    "peft",
 ]
 
 QUALITY_REQUIRES = [