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_models.py now downloads gfpgan model

Browse Source
This commit is contained in:
Lincoln Stein 2022-09-17 01:32:31 -04:00
parent 5b692f4720
commit 31daf1f0d7
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/preload_models.py
View File

@ -10,6 +10,7 @@ import sys
import transformers import transformers
import os import os
import warnings import warnings
import urllib.request
transformers.logging.set_verbosity_error() transformers.logging.set_verbosity_error()
@ -81,6 +82,16 @@ if gfpgan:
print('...success') print('...success')
except Exception: except Exception:
import traceback import traceback
print('Error loading ESRGAN:')
print(traceback.format_exc())
try:
import urllib.request
model_path = 'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth'
model_dest = 'src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth'
print('downloading gfpgan model file...')
urllib.request.urlretrieve(model_path,model_dest)
except Exception:
import traceback
print('Error loading GFPGAN:') print('Error loading GFPGAN:')
print(traceback.format_exc()) print(traceback.format_exc())