Improve the installation of optimum-neuron through optimum extras (#1778

) * pin for neuron * remove accelerate
huggingface · Apr 8, 2024 · 2ac676d · 2ac676d
1 parent 69f604c
commit 2ac676d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -81,8 +81,8 @@
     "neural-compressor": "optimum-intel[neural-compressor]>=1.15.0",
     "graphcore": "optimum-graphcore",
     "habana": ["optimum-habana", "transformers >= 4.37.0, < 4.38.0"],
-    "neuron": "optimum-neuron[neuron]",
-    "neuronx": "optimum-neuron[neuronx]",
+    "neuron": ["optimum-neuron[neuron]>=0.0.20", "transformers == 4.36.2"],
+    "neuronx": ["optimum-neuron[neuronx]>=0.0.20", "transformers == 4.36.2"],
     "furiosa": "optimum-furiosa",
     "amd": "optimum-amd",
     "dev": TESTS_REQUIRE + QUALITY_REQUIRE,