BeardedMonster commited on
Commit
31be57f
·
verified ·
1 Parent(s): 0a69dcf
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -76,8 +76,9 @@ generation_config = GenerationConfig(
76
  )
77
 
78
  repo_name = "BeardedMonster/SabiYarn-125M-finetune"
 
79
  model = AutoModelForCausalLM.from_pretrained(repo_name, trust_remote_code=True)
80
- tokenizer= AutoTokenizer.from_pretrained(repo_name, trust_remote_code=True)
81
 
82
  Use the following tags for the following downstream tasks:
83
  - Translation
 
76
  )
77
 
78
  repo_name = "BeardedMonster/SabiYarn-125M-finetune"
79
+ tokenizer_name = "BeardedMonster/SabiYarn-125M"
80
  model = AutoModelForCausalLM.from_pretrained(repo_name, trust_remote_code=True)
81
+ tokenizer= AutoTokenizer.from_pretrained(tokenizer_name, trust_remote_code=True)
82
 
83
  Use the following tags for the following downstream tasks:
84
  - Translation