From bc422239ade09edf1157ec9afd9a2416e1d6a940 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Fri, 22 Mar 2024 18:25:09 +0100
Subject: [PATCH] Update optimum version

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index d6e066b22c..dc5c417218 100644
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36.0,<4.40.0",
-    "optimum @ git+https://github.com/huggingface/optimum.git#egg=optimum",
+    "optimum~=1.18",
     "datasets>=1.4.0",
     "sentencepiece",
     "scipy",