github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch 'main' into dev

pull/41/head
duzx16 2023-03-14 20:34:04 +08:00
parent b271a928c0 076ea97423
commit 1e40acce6a
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -46,7 +46,7 @@ ChatGLM-6B 是一个开源的、支持中英双语的对话语言模型，基于
如果这些方法无法帮助你入睡,你可以考虑咨询医生或睡眠专家,寻求进一步的建议。 如果这些方法无法帮助你入睡,你可以考虑咨询医生或睡眠专家,寻求进一步的建议。
``` ```
完整的模型实现可以在 [HuggingFace Hub](https://huggingface.co/THUDM/chatglm-6b) 上查看。 完整的模型实现可以在 [Hugging Face Hub](https://huggingface.co/THUDM/chatglm-6b) 上查看。
### Demo ### Demo

4
web_demo.py
View File

@ -9,7 +9,9 @@ MAX_TURNS = 20
MAX_BOXES = MAX_TURNS * 2 MAX_BOXES = MAX_TURNS * 2
def predict(input, history=[]): def predict(input, history=None):
if history is None:
history = []
response, history = model.chat(tokenizer, input, history) response, history = model.chat(tokenizer, input, history)
updates = [] updates = []
for query, response in history: for query, response in history: