install from source

echarlaix · echarlaix · commit ddd35be27fd1 · 2024-05-16T16:36:03.000+02:00
diff --git a/setup.py b/setup.py
@@ -29,7 +29,8 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers @ git+https://github.com/huggingface/transformers.git",
-    "optimum~=1.19",
+    # "optimum~=1.19",
+    "optimum @ git+https://github.com/huggingface/optimum.git@bump-transformers",
     "datasets>=1.4.0",
     "sentencepiece",
     "scipy",
