From 93a01edd2ab7d3aedc13185fa933fffb4283901b Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Fri, 16 Feb 2024 09:46:28 +0100
Subject: [PATCH] Update setup

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index d7335bc93d..5edb7ee8ae 100644
--- 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",