From 83d2d55c94853d118a6a52e20fa8a0172c758143 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Fri, 26 Feb 2021 10:35:36 +0100 Subject: [PATCH] [ci, flax] non-existing models are unlikely to pass tests (#10409) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 😂 --- tests/test_flax_auto.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_flax_auto.py b/tests/test_flax_auto.py index 7d91b9009b..41c5d0d796 100644 --- a/tests/test_flax_auto.py +++ b/tests/test_flax_auto.py @@ -41,7 +41,7 @@ class FlaxAutoModelTest(unittest.TestCase): @slow def test_roberta_from_pretrained(self): - for model_name in ["roberta-base-cased", "roberta-large-uncased"]: + for model_name in ["roberta-base", "roberta-large"]: with self.subTest(model_name): config = AutoConfig.from_pretrained(model_name) self.assertIsNotNone(config) @@ -66,7 +66,7 @@ class FlaxAutoModelTest(unittest.TestCase): @slow def test_roberta_jax_jit(self): - for model_name in ["roberta-base-cased", "roberta-large-uncased"]: + for model_name in ["roberta-base", "roberta-large"]: tokenizer = AutoTokenizer.from_pretrained(model_name) model = FlaxRobertaModel.from_pretrained(model_name) tokens = tokenizer("Do you support jax jitted function?", return_tensors=TensorType.JAX)