AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Replace character name placeholders in chat context (closes #5007)

Browse Source
This commit is contained in:
oobabooga 2023-12-19 17:31:46 -08:00
parent e83e6cedbe
commit d8279dc710
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/chat.py
View File

@ -95,7 +95,8 @@ def generate_chat_prompt(user_input, state, **kwargs):
else: else:
renderer = chat_renderer renderer = chat_renderer
if state['context'].strip() != '': if state['context'].strip() != '':
messages.append({"role": "system", "content": state['context']}) context = replace_character_names(state['context'], state['name1'], state['name2'])
messages.append({"role": "system", "content": context})
insert_pos = len(messages) insert_pos = len(messages)
for user_msg, assistant_msg in reversed(history): for user_msg, assistant_msg in reversed(history):