From f2ba62b38c6dc7be7d50969ddc753fa3204b8fbc Mon Sep 17 00:00:00 2001 From: regisss <15324346+regisss@users.noreply.github.com> Date: Thu, 9 May 2024 12:58:50 +0200 Subject: [PATCH] Update the Transformers dependency in the Habana extra (#1851) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 7fac0a1c9d..4e154819bc 100644 --- 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",