marsggbo commited on
Commit
69e6d9a
1 Parent(s): f8944df

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-predictor_switch64_wmt16'
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