Fix custom tokenizers test (#19052)

* Fix CI for custom tokenizers

* Add nightly tests

* Run CI, run!

* Fix paths

* Typos

* Fix test
This commit is contained in:
Sylvain Gugger
2022-09-15 11:31:09 -04:00
committed by GitHub
parent 68bb33d770
commit f7ce4f1ff7
3 changed files with 74 additions and 6 deletions

View File

@@ -45,6 +45,7 @@ from transformers import (
SpecialTokensMixin,
Trainer,
TrainingArguments,
is_flax_available,
is_tf_available,
is_tokenizers_available,
is_torch_available,
@@ -2928,8 +2929,10 @@ class TokenizerTesterMixin:
returned_tensor = "pt"
elif is_tf_available():
returned_tensor = "tf"
else:
elif is_flax_available():
returned_tensor = "jax"
else:
return
if not tokenizer.pad_token or tokenizer.pad_token_id < 0:
return