This commit is contained in:
Saifeddine ALOUI 2023-05-30 00:55:45 +02:00
parent 8eedda84d6
commit b396d57e38

View File

@ -339,7 +339,7 @@ class ModelProcess:
prompt,
new_text_callback=callback,
n_predict=n_predict,
temp=self.config['temperature'],
temperature=self.config['temperature'],
top_k=self.config['top_k'],
top_p=self.config['top_p'],
repeat_penalty=self.config['repeat_penalty'],
@ -352,7 +352,7 @@ class ModelProcess:
prompt,
new_text_callback=callback,
n_predict=self.n_predict,
temp=self.personality.model_temperature,
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,