update setup

huggingface · Jul 5, 2024 · 52c3fc0 · 52c3fc0
1 parent 65e1b04
commit 52c3fc0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,8 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36.0,<4.43.0",
-    "optimum>=1.21.2,<1.22.0",
+    "optimum~=1.21",
+    # "optimum>=1.21.2,<1.22.0",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",