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

Fix default arguments of web demo

pull/41/head
duzx16 2023-03-14 20:30:25 +08:00
parent 0c6d1750ef
commit 584be64161
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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: