Update README.md
Browse files
README.md
CHANGED
@@ -9,7 +9,7 @@ import os
|
|
9 |
from glob import glob
|
10 |
from transformers import AutoModelForSeq2SeqLM, AutoConfig
|
11 |
|
12 |
-
model_name = 'marsggbo/t5-small_dff2048_dmodel32_token-pattern-
|
13 |
# ignore the mismatched size, because lm_head was modified
|
14 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
15 |
model_name, ignore_mismatched_sizes=True, use_safetensors=False
|
|
|
9 |
from glob import glob
|
10 |
from transformers import AutoModelForSeq2SeqLM, AutoConfig
|
11 |
|
12 |
+
model_name = 'marsggbo/t5-small_dff2048_dmodel32_token-pattern-predictor_switch64_xsum'
|
13 |
# ignore the mismatched size, because lm_head was modified
|
14 |
model = AutoModelForSeq2SeqLM.from_pretrained(
|
15 |
model_name, ignore_mismatched_sizes=True, use_safetensors=False
|