mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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:
renderer = chat_renderer
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)
for user_msg, assistant_msg in reversed(history):