File size: 298 Bytes
6d55e5f
 
 
 
9d4d56d
 
864f8bb
1
2
3
4
5
6
7
---
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)