Transformers upgrade (mosaicml#2489)

mvpatel2000 · Aug 29, 2023 · abfe27b · abfe27b
1 parent 23aaaeb
commit abfe27b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -184,7 +184,7 @@ def package_files(prefix: str, directory: str, extension: str):
 ]
 
 extra_deps['nlp'] = [
-    'transformers>=4.11,<4.32',
+    'transformers>=4.11,<4.33',
     'datasets>=2.4,<3',
 ]
 

diff --git a/tests/datasets/test_in_context_learning_datasets.py b/tests/datasets/test_in_context_learning_datasets.py
@@ -1264,6 +1264,7 @@ def test_code_eval_sentpiece_evaluation(monkeypatch, device, world_size, num_few
 @world_size(1, 2)
 @pytest.mark.parametrize('num_fewshot', [0, 2])
 @pytest.mark.parametrize('generations_per_sample', [1])
+@pytest.mark.filterwarnings(r'ignore: Input length of input_ids is')
 def test_code_eval_task_evaluation(monkeypatch, device, world_size, num_fewshot, dataset_uri, tiny_gpt2_tokenizer,
                                    tiny_gpt2_model, tmp_path, generations_per_sample):
     pytest.importorskip('datasets')