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

option to unload GFPGAN after using

Browse Source
This commit is contained in:
AUTOMATIC
2022-09-03 17:28:30 +03:00
parent f40617d6c4
commit 595c827bd3
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
modules/gfpgan_model.py
View File

@@ -4,6 +4,7 @@ import traceback
from modules.paths import script_path
from modules.shared import cmd_opts
import modules.shared
def gfpgan_model_path():
@@ -23,10 +24,18 @@ loaded_gfpgan_model = None
def gfpgan():
global loaded_gfpgan_model
if loaded_gfpgan_model is None and gfpgan_constructor is not None:
loaded_gfpgan_model = gfpgan_constructor(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
if loaded_gfpgan_model is not None:
return loaded_gfpgan_model
return loaded_gfpgan_model
if gfpgan_constructor is None:
return None
model = gfpgan_constructor(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
if not cmd_opts.unload_gfpgan:
loaded_gfpgan_model = model
return model
def gfpgan_fix_faces(np_image):