Archives/gpt4all
2
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all synced 2024-11-18 03:25:46 +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 59d7db9aad
commit f96300534b
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"
# Validate download directory
if model_path == None:
if model_path is None:
try:
os.makedirs(DEFAULT_MODEL_DIRECTORY, exist_ok=True)
except OSError as exc:
raise ValueError("Failed to create model download directory at {}: {}. "
"Please specify model_path.".format(DEFAULT_MODEL_DIRECTORY, exc))
model_path = DEFAULT_MODEL_DIRECTORY
if not os.path.exists(DEFAULT_MODEL_DIRECTORY):
try:
os.makedirs(DEFAULT_MODEL_DIRECTORY)
except:
raise ValueError("Failed to create model download directory at ~/.cache/gpt4all/. \
Please specify download_dir.")
else:
model_path = model_path.replace("\\", "\\\\")