github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-10-31 13:51:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Supports downloading officially supported models not hosted on gpt4all R2

Browse Source
This commit is contained in:
Andriy Mulyar
2023-06-06 16:09:16 -04:00
committed by Richard Guo
parent 266f13aee9
commit ef35eb496f
2 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -93,14 +93,23 @@ class GPT4All():
elif allow_download:
# Make sure valid model filename before attempting download
available_models = GPT4All.list_models()
if model_filename not in (m["filename"] for m in available_models):
selected_model = None
for m in available_models:
if model_filename == m['filename']:
selected_model = m
break
if selected_model is None:
raise ValueError(f"Model filename not in model list: {model_filename}")
return GPT4All.download_model(model_filename, model_path, verbose = verbose)
url = selected_model.pop('url', None)
return GPT4All.download_model(model_filename, model_path, verbose = verbose, url=url)
else:
raise ValueError("Failed to retrieve model")
@staticmethod
def download_model(model_filename: str, model_path: str, verbose: bool = True) -> str:
def download_model(model_filename: str, model_path: str, verbose: bool = True, url: str = None) -> str:
"""
Download model from https://gpt4all.io.
@@ -108,12 +117,15 @@ class GPT4All():
model_filename: Filename of model (with .bin extension).
model_path: Path to download model to.
verbose: If True (default), print debug messages.
url: the models remote url (e.g. may be hosted on HF)
Returns:
Model file destination.
"""
def get_download_url(model_filename):
if url:
return url
return f"https://gpt4all.io/models/{model_filename}"
# Download model