From 37d52c96bc5fac478d987f05a34d155e7cf2d97a Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 11 Apr 2023 21:46:17 -0300
Subject: [PATCH] Fix Continue in chat mode

---
 modules/chat.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/modules/chat.py b/modules/chat.py
index 428e8430..e922e7c3 100644
--- a/modules/chat.py
+++ b/modules/chat.py
@@ -139,10 +139,11 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False):
         text = apply_extensions(text, "input")
 
     # Generating the prompt
+    kwargs = {'_continue': _continue}
     if custom_generate_chat_prompt is None:
-        prompt = generate_chat_prompt(text, state)
+        prompt = generate_chat_prompt(text, state, **kwargs)
     else:
-        prompt = custom_generate_chat_prompt(text, state)
+        prompt = custom_generate_chat_prompt(text, state, **kwargs)
 
     # Yield *Is typing...*
     if not any((regenerate, _continue)):