CadenzaBaron commited on
Commit
ed5f209
·
1 Parent(s): a4f0c54

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -23,8 +23,8 @@ Sample generation script :
23
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
24
  import torch
25
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
26
- tokenizer = transformers.AutoTokenizer.from_pretrained(r"path\to\checkpoint")
27
- model = AutoModelForSeq2SeqLM.from_pretrained(r"path\to\checkpoint")
28
  model.to(device)
29
  tokenizer.src_lang = "zh"
30
  tokenizer.tgt_lang = "en"
 
23
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
24
  import torch
25
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
26
+ tokenizer = transformers.AutoTokenizer.from_pretrained("CadenzaBaron/M2M100-418M-for-GameTranslation-Finetuned-Zh-En")
27
+ model = AutoModelForSeq2SeqLM.from_pretrained("CadenzaBaron/M2M100-418M-for-GameTranslation-Finetuned-Zh-En")
28
  model.to(device)
29
  tokenizer.src_lang = "zh"
30
  tokenizer.tgt_lang = "en"