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

Fixes #2326, KeyError: 'assistant' (#2382)

Browse Source
This commit is contained in:
matatonic 2023-05-29 13:19:57 -04:00 committed by GitHub
parent 6de727c524
commit 2b7ba9586f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -259,7 +259,7 @@ class Handler(BaseHTTPRequestHandler):
role_formats = { role_formats = {
'user': 'user: {message}\n', 'user': 'user: {message}\n',
'bot': 'assistant: {message}\n', 'assistant': 'assistant: {message}\n',
'system': '{message}', 'system': '{message}',
'context': 'You are a helpful assistant. Answer as concisely as possible.', 'context': 'You are a helpful assistant. Answer as concisely as possible.',
'prompt': 'assistant:', 'prompt': 'assistant:',