Update predict.py

This commit is contained in:
binary-husky 2023-03-23 22:13:09 +08:00 committed by GitHub
parent 1b6d611888
commit a0f73a61ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -194,7 +194,7 @@ def predict(inputs, top_p, temperature, chatbot=[], history=[], system_prompt=''
chunk = get_full_error(chunk, stream_response)
error_msg = chunk.decode()
if "reduce the length" in error_msg:
chatbot[-1] = (history[-1], "老铁,输入的文本太长了")
chatbot[-1] = (history[-1], "输入的文本太长了超过gpt的令牌数量限制")
yield chatbot, history, "Json解析不合常规很可能是文本过长" + error_msg
return