From 98a87f3845aa19d0dee53767c448533ac93287b6 Mon Sep 17 00:00:00 2001
From: Daniel King <43149077+dakinggg@users.noreply.github.com>
Date: Thu, 1 Aug 2024 09:54:12 -0700
Subject: [PATCH] bump transformers version (#3511)

---
 pyproject.toml | 2 ++
 setup.py       | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 18f23ad267..7c9e35de15 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -160,6 +160,8 @@ filterwarnings = [
     '''ignore:.*an autograd kernel was not registered to the Autograd key.*:UserWarning''',
     # Ignore save_state_dict / load_state_dict deprecation warnings
     '''ignore:'.*_state_dict' is deprecated and will be removed in future versions.*:UserWarning''',
+    # Ignore mlflow warnings about transformers versions,
+    '''ignore:The 'transformers' MLflow Models integration.*:UserWarning''',
 ]
 
 # Coverage
diff --git a/setup.py b/setup.py
index eccba856fd..74877a82d2 100644
--- a/setup.py
+++ b/setup.py
@@ -179,7 +179,7 @@ def package_files(prefix: str, directory: str, extension: str):
 ]
 
 extra_deps['nlp'] = [
-    'transformers>=4.11,!=4.34.0,<4.43',
+    'transformers>=4.11,!=4.34.0,<4.44',
     'datasets>=2.4,<3',
     'huggingface-hub>=0.21.2,<0.24',
 ]