Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

preload script more robust for downloading gfpgan

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-05 11:29:48 -04:00
parent 3feda31d82
commit 488890e6bb
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/preload_models.py
View File

@ -72,6 +72,7 @@ if gfpgan:
if not os.path.exists(model_dest): if not os.path.exists(model_dest):
print('downloading gfpgan model file...') print('downloading gfpgan model file...')
os.makedirs(os.path.dirname(model_dest))
urllib.request.urlretrieve(model_url,model_dest) urllib.request.urlretrieve(model_url,model_dest)
except Exception: except Exception:
import traceback import traceback