From dc203e1e95f9416b1a60964a434049c33703a25f Mon Sep 17 00:00:00 2001 From: saloui Date: Thu, 22 Jun 2023 14:52:04 +0200 Subject: [PATCH] cancel gen repared --- api/__init__.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/api/__init__.py b/api/__init__.py index ae974fa0..463572d9 100644 --- a/api/__init__.py +++ b/api/__init__.py @@ -488,15 +488,12 @@ class LoLLMsAPPI(): ) self.socketio.sleep(0) self.current_discussion.update_message(self.current_ai_message_id, self.current_generated_text) - if self.cancel_gen: - ASCIIColors.warning("Generation canceled") - self.cancel_gen = False - - # if stop generation is detected then stop if not self.cancel_gen: return True else: + self.cancel_gen = False + ASCIIColors.warning("Generation canceled") return False else: self.current_generated_text = self.remove_text_from_string(self.current_generated_text, anti_prompt_to_remove)