fix setup

echarlaix · echarlaix · commit fddca6beec64 · 2024-05-22T15:40:59.000+02:00
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36.0,<4.41.0",
-    "optimum",
+    "optimum~=1.19",
     "datasets>=1.4.0",
     "sentencepiece",
     "scipy",
