[MT5] Fix CONFIG_MAPPING issue leading it to load umt5 class (#24678)

* update

* add umt5 to auto tokenizer mapping

* nits

* fixup

* fix failing torch test
This commit is contained in:
Arthur
2023-07-07 11:33:54 +09:00
committed by GitHub
parent fded6f4186
commit fb78769b9c
10 changed files with 212 additions and 15 deletions

View File

@@ -35,7 +35,7 @@ if is_torch_available():
from transformers import AutoTokenizer, UMT5ForConditionalGeneration, UMT5ForQuestionAnswering, UMT5Model
# Copied from test.models.t5.test_modeling_t5.T5ModelTester with T5->UMT5,UMT5Config->T5Config
# Copied from test.models.t5.test_modeling_t5.T5ModelTester with T5->UMT5
class UMT5ModelTester:
def __init__(
self,