fix

huggingface · Oct 9, 2024 · 3e35269 · 3e35269
1 parent 342b52e
commit 3e35269
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36,<4.46",
-    "optimum@git+https://github.com/huggingface/optimum.git@update-diffusers-mixins",
+    "optimum@git+https://github.com/huggingface/optimum.git",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",