bump transformers version (#3511)

mosaicml · Aug 1, 2024 · 98a87f3 · 98a87f3
1 parent 8390286
commit 98a87f3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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',
 ]