Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-29 10:59:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix stopping strings in the chat API

Browse Source
This commit is contained in:
oobabooga 2023-08-28 19:40:11 -07:00
parent 86c45b67ca
commit 439dd0faab
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/chat.py
View File

@ -176,6 +176,9 @@ def get_stopping_strings(state):
f"\n{state['name2']}:" f"\n{state['name2']}:"
] ]
if 'stopping_strings' in state and isinstance(state['stopping_strings'], list):
stopping_strings += state.pop('stopping_strings')
return stopping_strings return stopping_strings