diff --git a/setup.py b/setup.py
index 663528ef9c..cf4bfb3191 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "optimum@git+https://github.com/huggingface/optimum.git",
+    "optimum~=1.24",
     "transformers>=4.36,<4.48",
     "datasets>=1.4.0",
     "sentencepiece",