StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

load GFPGAN once and keep it in memory for #53

Browse Source
This commit is contained in:
AUTOMATIC 2022-09-03 09:05:48 +03:00
parent e5fc7bb59c
commit d7b67d9b40
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
webui.py
View File

@ -179,8 +179,16 @@ def gfpgan_model_path():
return found[0]
loaded_gfpgan_model = None
def gfpgan():
return GFPGANer(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
global loaded_gfpgan_model
if loaded_gfpgan_model is None:
loaded_gfpgan_model = GFPGANer(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
return loaded_gfpgan_model
def gfpgan_fix_faces(gfpgan_model, np_image):
np_image_bgr = np_image[:, :, ::-1]