ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-24 10:44:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed code

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-12-05 22:31:58 +01:00
parent dc0c6a69b4
commit 8388d197d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/personality.py
View File

@ -2179,7 +2179,7 @@ Do not answer with an id outside this possible answers.
["previous_discussion"] ["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>","") 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) self.print_prompt("Multi choice selection",prompt+gen)
try: try:
return int(selection) return int(selection)