github/home-llm
1
1
Fork 0
mirror of https://github.com/acon96/home-llm.git synced 2026-01-08 21:28:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

handle failure to create models directory

Browse Source
This commit is contained in:
Alex O'Connell
2024-04-16 21:17:22 -04:00
parent 556edc19a7
commit 33a6cd96ed
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
custom_components/llama_conversation/utils.py
View File

@@ -54,7 +54,10 @@ def download_model_from_hf(model_name: str, quantization_type: str, storage_fold
if len(wanted_file) != 1:
raise Exception(f"The quantization '{quantization_type}' does not exist in the HF repo for {model_name}")
os.makedirs(storage_folder, exist_ok=True)
try:
os.makedirs(storage_folder, exist_ok=True)
except Exception as ex:
raise Exception(f"Failed to create the required folder for storing models! You may need to create the path '{storage_folder}' manually.") from ex
return hf_hub_download(
repo_id=model_name,