Update the Transformers dependency in the Habana extra (#1851)

huggingface · May 9, 2024 · b3ecb6c · b3ecb6c
1 parent db6db6f
commit b3ecb6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -80,7 +80,7 @@
     "nncf": "optimum-intel[nncf]>=1.15.0",
     "neural-compressor": "optimum-intel[neural-compressor]>=1.15.0",
     "graphcore": "optimum-graphcore",
-    "habana": ["optimum-habana", "transformers >= 4.37.0, < 4.38.0"],
+    "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"],
     "furiosa": "optimum-furiosa",