Commit ·
864f8bb
1
Parent(s): 6d55e5f
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,4 +2,6 @@
|
|
| 2 |
language:
|
| 3 |
- zh
|
| 4 |
---
|
| 5 |
-
tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
|
|
|
|
|
|
|
|
|
| 2 |
language:
|
| 3 |
- zh
|
| 4 |
---
|
| 5 |
+
tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
|
| 6 |
+
config = AutoConfig.from_pretrained("Leonardolin/mt5-NTCIR17-3classification")
|
| 7 |
+
model = MT5ForConditionalGeneration.from_pretrained("Leonardolin/mt5-NTCIR17-3classification", config = config,).to(device)
|