ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-01-31 08:25:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix in gpt-j model

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-04-20 19:45:35 +02:00
parent 7472ccef19
commit ac3f6e430f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyGpt4All/backends/gpt_j.py
View File

@ -48,7 +48,7 @@ class GPT_J(GPTBackend):
"""
self.model.generate(
prompt,
new_text_callback=new_text_callback,
#new_text_callback=new_text_callback,
n_predict=n_predict,
temp=self.config['temp'],
top_k=self.config['top_k'],
@ -58,3 +58,4 @@ class GPT_J(GPTBackend):
n_threads=self.config['n_threads'],
verbose=verbose
)
new_text_callback()