AI/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-09-19 15:25:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Concise model matching

Browse Source
This commit is contained in:
Konstantin Gukov 2023-05-25 17:29:13 +02:00 committed by Richard Guo
parent c1f3dd310c
commit a067f38544
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -96,12 +96,8 @@ class GPT4All():
# If model file does not exist, download # If model file does not exist, download
elif allow_download: elif allow_download:
# Make sure valid model filename before attempting download # Make sure valid model filename before attempting download
model_match = False available_models = GPT4All.list_models()
for item in GPT4All.list_models(): if model_filename not in (m["filename"] for m in available_models):
if model_filename == item["filename"]:
model_match = True
break
if not model_match:
raise ValueError(f"Model filename not in model list: {model_filename}") raise ValueError(f"Model filename not in model list: {model_filename}")
return GPT4All.download_model(model_filename, model_path) return GPT4All.download_model(model_filename, model_path)
else: else: