Update optimum neuron extra (#2190)

* feat: remove neuron extra * feat: remove transformers from neuronx extra
huggingface · Feb 17, 2025 · 27dae50 · 27dae50
1 parent 6335599
commit 27dae50
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -99,8 +99,7 @@
     "neural-compressor": "optimum-intel[neural-compressor]>=1.18.0",
     "ipex": "optimum-intel[ipex]>=1.18.0",
     "habana": ["optimum-habana", "transformers>=4.45.0,<4.46.0"],
-    "neuron": ["optimum-neuron[neuron]>=0.0.20", "transformers>=4.36.2,<4.42.0"],
-    "neuronx": ["optimum-neuron[neuronx]>=0.0.20", "transformers>=4.36.2,<4.42.0"],
+    "neuronx": ["optimum-neuron[neuronx]>=0.0.28"],
     "graphcore": "optimum-graphcore",
     "furiosa": "optimum-furiosa",
     "amd": "optimum-amd",