Bump transformers version for neuron extras (#1881)

huggingface · May 29, 2024 · a2562e6 · a2562e6
1 parent e9af0f2
commit a2562e6
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.16.0",
     "graphcore": "optimum-graphcore",
     "habana": ["optimum-habana", "transformers >= 4.38.0, < 4.39.0"],
-    "neuron": ["optimum-neuron[neuron]>=0.0.20", "transformers == 4.36.2"],
-    "neuronx": ["optimum-neuron[neuronx]>=0.0.20", "transformers == 4.36.2"],
+    "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"],
     "furiosa": "optimum-furiosa",
     "amd": "optimum-amd",
     "dev": TESTS_REQUIRE + QUALITY_REQUIRE,