[ci, flax] non-existing models are unlikely to pass tests (#10409)

😂
huggingface · Feb 26, 2021 · 83d2d55 · 83d2d55
1 parent 17b6e0d
commit 83d2d55
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_flax_auto.py b/tests/test_flax_auto.py
@@ -41,7 +41,7 @@ def test_bert_from_pretrained(self):
 
     @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 @@ def eval(**kwargs):
 
     @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)