michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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]