From 2d134031ca4f14230ce923a974f7795c2cf4857f Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sun, 29 Jan 2023 00:04:11 -0300 Subject: [PATCH] Apply extensions to character greeting --- server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server.py b/server.py index 9f1af694..f468a176 100644 --- a/server.py +++ b/server.py @@ -499,8 +499,8 @@ if args.chat or args.cai_chat: if 'example_dialogue' in data and data['example_dialogue'] != '': history['internal'] = tokenize_dialogue(data['example_dialogue'], name1, name2) if 'char_greeting' in data and len(data['char_greeting'].strip()) > 0: - history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', data['char_greeting']]] - history['visible'] += [['', data['char_greeting']]] + history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', apply_extensions(data['char_greeting'], "output")]] + history['visible'] += [['', apply_extensions(data['char_greeting'], "output")]] else: history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', "Hello there!"]] history['visible'] += [['', "Hello there!"]]