mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-01 01:26:03 -04:00
Handle <USER> and <BOT>
This commit is contained in:
parent
626da6c731
commit
19ccb2aaf5
@ -332,6 +332,7 @@ def load_character(_character, name1, name2):
|
||||
context = f"{context.strip()}\n<START>\n"
|
||||
if 'example_dialogue' in data and data['example_dialogue'] != '':
|
||||
data['example_dialogue'] = data['example_dialogue'].replace('{{user}}', name1).replace('{{char}}', name2)
|
||||
data['example_dialogue'] = data['example_dialogue'].replace('<USER>', name1).replace('<BOT>', name2)
|
||||
context += f"{data['example_dialogue'].strip()}\n"
|
||||
if 'char_greeting' in data and len(data['char_greeting'].strip()) > 0:
|
||||
shared.history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', data['char_greeting']]]
|
||||
|
Loading…
Reference in New Issue
Block a user