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

Fix turn_idx in eval

pull/573/head
duzx16 2023-04-12 23:41:37 +08:00
parent f06df225dd
commit 7607cfe585
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ptuning/main.py
View File

@ -166,8 +166,8 @@ def main():
else: else:
prompt = "" prompt = ""
history = examples[history_column][i] history = examples[history_column][i]
for i, (old_query, response) in enumerate(history): for turn_idx, (old_query, response) in enumerate(history):
prompt += "[Round {}]\n问：{}\n答：{}\n".format(i, old_query, response) prompt += "[Round {}]\n问：{}\n答：{}\n".format(turn_idx, old_query, response)
prompt += "[Round {}]\n问：{}\n答：".format(len(history), query) prompt += "[Round {}]\n问：{}\n答：".format(len(history), query)
inputs.append(prompt) inputs.append(prompt)
targets.append(examples[response_column][i]) targets.append(examples[response_column][i])