Adds pretrained IDs directly in the tests (#29534)

* Adds pretrained IDs directly in the tests * Fix tests * Fix tests * Review!
2024-03-13 14:53:27 +01:00
parent 38bff8c84f
commit 11bbb505c7
89 changed files with 95 additions and 8 deletions
--- a/tests/models/distilbert/test_tokenization_distilbert.py
+++ b/tests/models/distilbert/test_tokenization_distilbert.py
@@ -25,6 +25,7 @@ class DistilBertTokenizationTest(BertTokenizationTest):
    tokenizer_class = DistilBertTokenizer
    rust_tokenizer_class = DistilBertTokenizerFast
    test_rust_tokenizer = True
+    from_pretrained_id = "distilbert/distilbert-base-uncased"

    @slow
    def test_sequence_builders(self):