ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-01-27 14:50:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update personality.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-09-04 13:58:11 +02:00 committed by GitHub
parent add8b43323
commit 86238fe666
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lollms/personality.py
View File

@ -749,7 +749,7 @@ class AIPersonality:
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
def generate(self, prompt, max_size = None, temperature = None, top_k = None, top_p=None, repeat_penalty=None, repeat_last_n=None, callback=None, debug=False, show_progress=False ):
@ -769,7 +769,7 @@ class AIPersonality:
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()
)
if debug:
self.print_prompt("prompt", prompt+self.bot_says)
@ -3935,7 +3935,7 @@ class APScript(StateMachine):
return rounds_info
def answer(self, context_details, callback=None, send_full=True, custom_entries = ""):
def answer(self, context_details, custom_entries = "", send_full=True, callback=None):
if context_details["is_continue"]:
full_prompt = self.build_prompt_from_context_details(context_details, custom_entries=custom_entries, suppress= ["ai_prefix"])
else: