diff --git a/setup.py b/setup.py index d132975aa4..f291ed7125 100644 --- 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",