github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-06-23 22:18:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Nicer handling of missing model directory.

Browse Source 
Correct exception message.
This commit is contained in:
Konstantin Gukov 2023-05-25 17:26:15 +02:00 committed by Richard Guo
parent 84201784b5
commit b6b21441c7
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -77,14 +77,13 @@ class GPT4All():
model_filename += ".bin" model_filename += ".bin"
# Validate download directory # Validate download directory
if model_path == None: if model_path is None:
model_path = DEFAULT_MODEL_DIRECTORY
if not os.path.exists(DEFAULT_MODEL_DIRECTORY):
try: try:
os.makedirs(DEFAULT_MODEL_DIRECTORY) os.makedirs(DEFAULT_MODEL_DIRECTORY, exist_ok=True)
except: except OSError as exc:
raise ValueError("Failed to create model download directory at ~/.cache/gpt4all/. \ raise ValueError("Failed to create model download directory at {}: {}. "
Please specify download_dir.") "Please specify model_path.".format(DEFAULT_MODEL_DIRECTORY, exc))
model_path = DEFAULT_MODEL_DIRECTORY
else: else:
model_path = model_path.replace("\\", "\\\\") model_path = model_path.replace("\\", "\\\\")