From 27b238999f292eeed0ba3853a395ef377ed72f31 Mon Sep 17 00:00:00 2001 From: Joseph Dries III Date: Mon, 19 Sep 2022 20:04:44 -0400 Subject: [PATCH] Use model_url instead of model_path for gfpgan and CodeFormer. Fixes lstein/stable-diffusion#678 --- scripts/preload_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/preload_models.py b/scripts/preload_models.py index 6c082d80e9..d19cc6d8d1 100644 --- a/scripts/preload_models.py +++ b/scripts/preload_models.py @@ -92,7 +92,7 @@ if gfpgan: if not os.path.exists(model_dest): print('downloading gfpgan model file...') - urllib.request.urlretrieve(model_path,model_dest) + urllib.request.urlretrieve(model_url,model_dest) except Exception: import traceback print('Error loading GFPGAN:') @@ -107,7 +107,7 @@ try: if not os.path.exists(model_dest): print('downloading codeformer model file...') os.makedirs(os.path.dirname(model_dest), exist_ok=True) - urllib.request.urlretrieve(model_path,model_dest) + urllib.request.urlretrieve(model_url,model_dest) except Exception: import traceback print('Error loading CodeFormer:')