From b1537b5df3ebefeaa81c147b9573b103bb4e5a4e Mon Sep 17 00:00:00 2001 From: saloui Date: Mon, 5 Feb 2024 16:52:54 +0100 Subject: [PATCH] fix --- lollms/personality.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/personality.py b/lollms/personality.py index f13b71a..0099210 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -2550,7 +2550,7 @@ The AI should respond in this format using data from actions_list: elements += ["!@>answer:"] prompt = self.build_prompt(elements) - gen = self.generate(prompt, max_answer_length, temperature=0.1, top_k=50, top_p=0.9, repeat_penalty=1.0, repeat_last_n=50).strip().replace("","").replace("","") + gen = self.generate(prompt, max_answer_length, temperature=0.1, top_k=50, top_p=0.9, repeat_penalty=1.0, repeat_last_n=50, callback=self.sink).strip().replace("","").replace("","") selection = gen.strip().split()[0].replace(",","").replace(".","") self.print_prompt("Multi choice selection",prompt+gen) try: