diff --git a/setup.py b/setup.py index 81b3d74c9c..7b555bbd88 100644 --- 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,