From a97d5e14d69fd62db095543dd6bdf0c22484d666 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Thu, 8 Feb 2024 15:13:00 +0100 Subject: [PATCH] Update personality.py --- lollms/personality.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lollms/personality.py b/lollms/personality.py index a14deb3..390d565 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -513,7 +513,7 @@ Date: {{date}} self.bot_says = bot_says return True - def generate_with_images(self, prompt, images, max_size, temperature = None, top_k = None, top_p=None, repeat_penalty=None, callback=None, debug=False, show_progress=False ): + def generate_with_images(self, prompt, images, max_size, temperature = None, top_k = None, top_p=None, repeat_penalty=None, repeat_last_n=None, callback=None, debug=False, show_progress=False ): ASCIIColors.info("Text generation started: Warming up") self.nb_received_tokens = 0 self.bot_says = "" @@ -529,6 +529,7 @@ Date: {{date}} top_k=self.model_top_k if top_k is None else top_k, top_p=self.model_top_p if top_p is None else top_p, repeat_penalty=self.model_repeat_penalty if repeat_penalty is None else repeat_penalty, + repeat_last_n = self.model_repeat_last_n if repeat_last_n is None else repeat_last_n ).strip() return self.bot_says