Fix Optimum dependency so that optimum[habana] is correctly installed

regisss · regisss · commit 8f02490282ad · 2025-10-21T16:41:25.000Z
diff --git a/setup.py b/setup.py
@@ -30,7 +30,7 @@
 
 INSTALL_REQUIRES = [
     "transformers >= 4.55.0, < 4.56.0",
-    "optimum ~= 1.25",
+    "optimum ~= 2.0",
     "torch",
     "accelerate >= 1.7.0",
     "diffusers >= 0.34.0, < 0.34.1",
