From 70bd6725bceaa29316297885280a58f0587231c1 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Fri, 16 Aug 2024 15:07:30 +0200
Subject: [PATCH] Update Habana extra in setup.py (#1991)

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 3043eeee6c8..2e8c9489a89 100644
--- a/setup.py
+++ b/setup.py
@@ -80,7 +80,7 @@
     "nncf": "optimum-intel[nncf]>=1.18.0",
     "neural-compressor": "optimum-intel[neural-compressor]>=1.18.0",
     "ipex": "optimum-intel[ipex]>=1.18.0",
-    "habana": ["optimum-habana", "transformers >= 4.40.0, < 4.41.0"],
+    "habana": ["optimum-habana", "transformers >= 4.43.0, < 4.44.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"],
     "graphcore": "optimum-graphcore",