Adds pretrained IDs directly in the tests (#29534)

* Adds pretrained IDs directly in the tests

* Fix tests

* Fix tests

* Review!
This commit is contained in:
Lysandre Debut
2024-03-13 14:53:27 +01:00
committed by GitHub
parent 38bff8c84f
commit 11bbb505c7
89 changed files with 95 additions and 8 deletions

View File

@@ -30,6 +30,7 @@ from ...test_tokenization_common import TokenizerTesterMixin
@require_tokenizers
# Copied from tests.models.roberta.test_tokenization_roberta.RobertaTokenizationTest with FacebookAI/roberta-base->allenai/longformer-base-4096,Roberta->Longformer,roberta->longformer,
class LongformerTokenizationTest(TokenizerTesterMixin, unittest.TestCase):
from_pretrained_id = "allenai/longformer-base-4096"
# Ignore copy
tokenizer_class = LongformerTokenizer
test_slow_tokenizer = True