Explorar o código

Handle <USER> and <BOT>

oobabooga %!s(int64=2) %!d(string=hai) anos
pai
achega
19ccb2aaf5
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      modules/chat.py

+ 1 - 0
modules/chat.py

@@ -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']]]