hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Cleanup existing directories, fixes

Browse Source
This commit is contained in:
d8ahazard
2022-09-26 10:27:18 -05:00
parent 740070ea9c
commit 7d5c29b674
6 changed files with 48 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/esrgan_model.py
View File

@@ -18,7 +18,7 @@ from modules.shared import opts
model_dir = "ESRGAN"
model_path = os.path.join(models_path, model_dir)
model_url = "https://drive.google.com/u/0/uc?id=1TPrz5QKd8DHHt1k8SRtm6tMiPjz_Qene&export=download"
model_name = "ESRGAN_x4.pth"
model_name = "ESRGAN_x4"
def load_model(path: str, name: str):
@@ -27,7 +27,7 @@ def load_model(path: str, name: str):
global model_dir
global model_name
if "http" in path:
filename = load_file_from_url(url=model_url, model_dir=model_path, file_name=model_name, progress=True)
filename = load_file_from_url(url=model_url, model_dir=model_path, file_name="%s.pth" % model_name, progress=True)
else:
filename = path
if not os.path.exists(filename) or filename is None: