fixed code

ParisNeo · Dec 5, 2023 · 8388d19 · 8388d19
1 parent dc0c6a6
commit 8388d19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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("</s>","").replace("<s>","")
-        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)