github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-07-16 08:26:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add optional verbosity

Browse Source
This commit is contained in:
Konstantin Gukov 2023-05-26 09:53:53 +02:00 committed by Richard Guo
parent e05ee9466a
commit 5e61008424
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
gpt4all-bindings/python/gpt4all/gpt4all.py
View File

@ -58,7 +58,7 @@ class GPT4All():
return requests.get("https://gpt4all.io/models/models.json").json() return requests.get("https://gpt4all.io/models/models.json").json()
@staticmethod @staticmethod
def retrieve_model(model_name: str, model_path: str = None, allow_download: bool = True) -> str: def retrieve_model(model_name: str, model_path: str = None, allow_download: bool = True, verbose: bool = True) -> str:
""" """
Find model file, and if it doesn't exist, download the model. Find model file, and if it doesn't exist, download the model.
@ -67,6 +67,7 @@ class GPT4All():
model_path: Path to find model. Default is None in which case path is set to model_path: Path to find model. Default is None in which case path is set to
~/.cache/gpt4all/. ~/.cache/gpt4all/.
allow_download: Allow API to download model from gpt4all.io. Default is True. allow_download: Allow API to download model from gpt4all.io. Default is True.
verbose: If True (default), print debug messages.
Returns: Returns:
Model file destination. Model file destination.
@ -92,7 +93,8 @@ class GPT4All():
model_dest = os.path.join(model_path, model_filename).replace("\\", "\\\\") model_dest = os.path.join(model_path, model_filename).replace("\\", "\\\\")
if os.path.exists(model_dest): if os.path.exists(model_dest):
print("Found model file at ", model_dest) if verbose:
print("Found model file at ", model_dest)
return model_dest return model_dest
# If model file does not exist, download # If model file does not exist, download
@ -106,13 +108,14 @@ class GPT4All():
raise ValueError("Failed to retrieve model") raise ValueError("Failed to retrieve model")
@staticmethod @staticmethod
def download_model(model_filename: str, model_path: str) -> str: def download_model(model_filename: str, model_path: str, verbose: bool) -> str:
""" """
Download model from https://gpt4all.io. Download model from https://gpt4all.io.
Args: Args:
model_filename: Filename of model (with .bin extension). model_filename: Filename of model (with .bin extension).
model_path: Path to download model to. model_path: Path to download model to.
verbose: If True (default), print debug messages.
Returns: Returns:
Model file destination. Model file destination.
@ -137,7 +140,8 @@ class GPT4All():
file.write(data) file.write(data)
except Exception: except Exception:
if os.path.exists(download_path): if os.path.exists(download_path):
print('Cleaning up the interrupted download...') if verbose:
print('Cleaning up the interrupted download...')
os.remove(download_path) os.remove(download_path)
raise raise
@ -150,7 +154,8 @@ class GPT4All():
# Sleep for a little bit so OS can remove file lock # Sleep for a little bit so OS can remove file lock
time.sleep(2) time.sleep(2)
print("Model downloaded at: " + download_path) if verbose:
print("Model downloaded at: ", download_path)
return download_path return download_path
def generate(self, prompt: str, streaming: bool = True, **generate_kwargs) -> str: def generate(self, prompt: str, streaming: bool = True, **generate_kwargs) -> str: