Limit optimum version (#807)

echarlaix · web-flow · commit 48cc82aa361b · 2024-07-05T15:07:23.000+02:00
* Limit optimum version

* trigger test

* trigger test
diff --git 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",
