From c6b57bfedd311a870e8dbbc0cba229f485db8ed7 Mon Sep 17 00:00:00 2001
From: liammcinroy <1899809+liammcinroy@users.noreply.github.com>
Date: Tue, 23 Jul 2024 14:57:05 -0600
Subject: [PATCH] Support transformers 4.43

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

diff --git a/setup.py b/setup.py
index 41598aeba5..1f811ebc7d 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
 REQUIRED_PKGS = [
     "coloredlogs",
     "sympy",
-    "transformers[sentencepiece]>=4.26.0,<4.43.0",
+    "transformers[sentencepiece]>=4.26.0,<4.44",
     "torch>=1.11",
     "packaging",
     "numpy<2.0",  # transformers requires numpy<2.0 https://github.com/huggingface/transformers/pull/31569