From 8388d197d2b72a42f270761fd4ba9fd7b757e321 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Tue, 5 Dec 2023 22:31:58 +0100 Subject: [PATCH] fixed code --- lollms/personality.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/personality.py b/lollms/personality.py index 5bfd239e..8a73c6dd 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -2179,7 +2179,7 @@ def multichoice_question(self, question: str, possible_answers:list, context:str ["previous_discussion"] ) 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("","") - selection = gen.strip().split()[0].replace(",","") + selection = gen.strip().split()[0].replace(",","").replace(".","") self.print_prompt("Multi choice selection",prompt+gen) try: return int(selection)