KaleiNeely commited on
Commit
371a19a
1 Parent(s): 2027efa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -130,7 +130,7 @@ User: {instruction}
130
  Assistant:"""
131
 
132
  model = AutoModelForCausalLM.from_pretrained("RWKV/rwkv-5-world-3b", trust_remote_code=True).to(torch.float32)
133
- tokenizer = AutoTokenizer.from_pretrained("RWKV/rwkv-5-world-3b", trust_remote_code=True, padding_side='left')
134
 
135
  texts = ["请介绍北京的旅游景点", "介绍一下大熊猫", "乌兰察布"]
136
  prompts = [generate_prompt(text) for text in texts]
 
130
  Assistant:"""
131
 
132
  model = AutoModelForCausalLM.from_pretrained("RWKV/rwkv-5-world-3b", trust_remote_code=True).to(torch.float32)
133
+ tokenizer = AutoTokenizer.from_pretrained("RWKV/rwkv-5-world-3b", trust_remote_code=True, padding_side='left', pad_token="<s>")
134
 
135
  texts = ["请介绍北京的旅游景点", "介绍一下大熊猫", "乌兰察布"]
136
  prompts = [generate_prompt(text) for text in texts]