From ab94f0d9bf0dddb8a7465bd40637f1e17e5dd832 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sat, 18 Nov 2023 21:11:04 -0800 Subject: [PATCH] Minor style change --- modules/chat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/chat.py b/modules/chat.py index 436e3e4..22b5bf9 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -96,6 +96,7 @@ def generate_chat_prompt(user_input, state, **kwargs): context_instruct = context_instruct.replace('<|system-message|>', state['custom_system_message']) else: context_instruct = context_instruct.replace('<|system-message|>', state['system_message']) + wrapper += context_instruct wrapper += all_substrings['instruct']['user_turn'].replace('<|user-message|>', command) wrapper += all_substrings['instruct']['bot_turn_stripped']