Update setup

huggingface · Feb 16, 2024 · 93a01ed · 93a01ed
1 parent 6c8fa79
commit 93a01ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "optimum @ git+https://github.com/huggingface/optimum.git",  # TODO : 1.17.0
+    "optimum>=1.17.0",
     "transformers>=4.26.0",
     "datasets>=1.4.0",
     "sentencepiece",