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

enhanced

Browse Source
This commit is contained in:
saloui 2023-06-27 16:47:47 +02:00
parent 17e51933d8
commit 4c405fa3d2
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lollms/personality.py
View File

@ -988,16 +988,17 @@ class APScript:
self.bot_says = bot_says
return True
def generate(self, prompt, max_size):
def generate(self, prompt, max_size, temperature = None, top_k = None, top_p=None, repeat_penalty=None ):
self.bot_says = ""
ASCIIColors.info("Text generation started: Warming up")
return self.personality.model.generate(
prompt,
max_size,
self.process,
temperature=self.personality.model_temperature,
top_k=self.personality.model_top_k,
top_p=self.personality.model_top_p,
repeat_penalty=self.personality.model_repeat_penalty,
temperature=self.personality.model_temperature if temperature is None else temperature,
top_k=self.personality.model_top_k if top_k is None else top_k,
top_p=self.personality.model_top_p if top_p is None else top_p,
repeat_penalty=self.personality.model_repeat_penalty if repeat_penalty is None else repeat_penalty,
).strip()
def run_workflow(self, prompt:str, previous_discussion_text:str="", callback=None):

2
setup.py
View File

@ -26,7 +26,7 @@ def get_all_files(path):
setuptools.setup(
name="lollms",
version="2.0.25",
version="2.0.27",
author="Saifeddine ALOUI",
author_email="aloui.saifeddine@gmail.com",
description="A python library for AI personality definition",