| language: | |
| - zh | |
| tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")</br> | |
| config = AutoConfig.from_pretrained("Leonardolin/mt5-NTCIR17-3classification") </br> | |
| model = MT5ForConditionalGeneration.from_pretrained("Leonardolin/mt5-NTCIR17-3classification", config = config,).to(device) |