chenxingphh
commited on
Commit
•
ba75c6d
1
Parent(s):
746d87c
Update README.md
Browse files
README.md
CHANGED
@@ -89,7 +89,7 @@ model = AutoModelForCausalLM.from_pretrained("OrionStarAI/OrionStar-Yi-34B-Chat"
|
|
89 |
|
90 |
model.generation_config = GenerationConfig.from_pretrained("OrionStarAI/OrionStar-Yi-34B-Chat")
|
91 |
messages = [{"role": "user", "content": "你好! 你叫什么名字!"}]
|
92 |
-
response = model.chat(tokenizer, messages, streaming=
|
93 |
print(response)
|
94 |
|
95 |
# 你好,我的名字叫聚言,很高兴见到你。
|
|
|
89 |
|
90 |
model.generation_config = GenerationConfig.from_pretrained("OrionStarAI/OrionStar-Yi-34B-Chat")
|
91 |
messages = [{"role": "user", "content": "你好! 你叫什么名字!"}]
|
92 |
+
response = model.chat(tokenizer, messages, streaming=False)
|
93 |
print(response)
|
94 |
|
95 |
# 你好,我的名字叫聚言,很高兴见到你。
|