From a2870c85a9228e464c34f5c86b2e43af46346911 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
Date: Fri, 5 Jul 2024 15:07:23 +0200
Subject: [PATCH] Limit optimum version (#807)

* Limit optimum version

* trigger test

* trigger test
---
 setup.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index a1c83259a6..8abcbfaf08 100644
--- a/setup.py
+++ b/setup.py
@@ -29,8 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36.0,<4.43.0",
-    "optimum~=1.21",
-    # "optimum>=1.21.2,<1.22.0",
+    "optimum>=1.21.2,<1.22.0",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",