KaleiNeely
commited on
Commit
•
371a19a
1
Parent(s):
2027efa
Update README.md
Browse files
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]
|