From b670912834fac7e03253f5bfd4e4993f74e19d2c Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Wed, 10 Jul 2024 17:24:52 +0200 Subject: [PATCH] Update personality.py --- lollms/personality.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lollms/personality.py b/lollms/personality.py index 2270aeb..65a8e94 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -1738,11 +1738,11 @@ class AIPersonality: Returns: bool: True if any antiprompt is found in the text (ignoring case), False otherwise. """ - start_header_id_template = self.config.start_header_id_template - end_header_id_template = self.config.end_header_id_template - separator_template = self.config.separator_template + start_header_id_template = self.config.start_header_id_template + start_user_header_id_template = self.config.start_user_header_id_template + start_ai_header_id_template = self.config.start_ai_header_id_template - anti_prompts = [start_header_id_template, self.app.config.discussion_prompt_separator] + anti_prompts = [start_header_id_template, start_user_header_id_template, start_ai_header_id_template] if self.app.config.separator_template!="\n": anti_prompts.append(self.app.config.separator_template)