diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-03 06:05:48 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-03 06:05:48 +0000 |
| commit | d7b67d9b40e47ede766d3beb149b0c2b74651ece (patch) | |
| tree | 4cd81e4b263d5b9dee4aed02a3ee3e114f715383 /webui.py | |
| parent | e5fc7bb59c311e137cc54d15f96e3d3537c07429 (diff) | |
| download | stable-diffusion-webui-gfx803-d7b67d9b40e47ede766d3beb149b0c2b74651ece.tar.gz stable-diffusion-webui-gfx803-d7b67d9b40e47ede766d3beb149b0c2b74651ece.tar.bz2 stable-diffusion-webui-gfx803-d7b67d9b40e47ede766d3beb149b0c2b74651ece.zip | |
load GFPGAN once and keep it in memory for #53
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 10 |
1 files changed, 9 insertions, 1 deletions
@@ -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]
|