Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add .rstrip(' ') to openai api

Browse Source
This commit is contained in:
oobabooga 2023-05-12 14:40:48 -03:00
parent 3f1bfba718
commit c746a5bd00
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/openai/script.py
View File

@ -70,7 +70,7 @@ def deduce_template():
.replace('<|user|>', instruct.get('user', ''))\ .replace('<|user|>', instruct.get('user', ''))\
.replace('<|bot|>', instruct.get('bot', ''))\ .replace('<|bot|>', instruct.get('bot', ''))\
.replace('<|user-message|>', '{instruction}\n{input}') .replace('<|user-message|>', '{instruction}\n{input}')
return instruct.get('context', '') + template[:template.find('<|bot-message|>')] return instruct.get('context', '') + template[:template.find('<|bot-message|>')].rstrip(' ')
except: except:
return default_template return default_template