mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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