diff --git a/setup.py b/setup.py
index e54d6626d7..d5f0db26d0 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@
 
 INSTALL_REQUIRES = [
     "transformers >= 4.49.0, < 4.50.0",
-    "optimum",
+    "optimum ~= 1.25",
     "torch",
     "accelerate >= 1.7.0",
     "diffusers >= 0.33.1, < 0.33.2",