diff options
| author | Jairo Correa <jn.j41r0@gmail.com> | 2022-09-29 01:14:13 +0000 |
|---|---|---|
| committer | Jairo Correa <jn.j41r0@gmail.com> | 2022-09-29 01:14:13 +0000 |
| commit | c938679de7b87b4f14894d9f57fe0f40dd6e3c06 (patch) | |
| tree | 31119c9dc5d04648bf873069c690b4fd53dc3805 /modules/gfpgan_model.py | |
| parent | 041d2aefc082c2883aa7e28ee3e4a990b3be9758 (diff) | |
| download | stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.tar.gz stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.tar.bz2 stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.zip | |
Fix memory leak and reduce memory usage
Diffstat (limited to 'modules/gfpgan_model.py')
| -rw-r--r-- | modules/gfpgan_model.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py index 44c5dc6c..b1288f0c 100644 --- a/modules/gfpgan_model.py +++ b/modules/gfpgan_model.py @@ -49,6 +49,7 @@ def gfpgan(): def gfpgan_fix_faces(np_image):
+ global loaded_gfpgan_model
model = gfpgan()
np_image_bgr = np_image[:, :, ::-1]
@@ -56,7 +57,9 @@ def gfpgan_fix_faces(np_image): np_image = gfpgan_output_bgr[:, :, ::-1]
if shared.opts.face_restoration_unload:
- model.gfpgan.to(devices.cpu)
+ del model
+ loaded_gfpgan_model = None
+ devices.torch_gc()
return np_image
@@ -83,11 +86,7 @@ def setup_gfpgan(): return "GFPGAN"
def restore(self, np_image):
- np_image_bgr = np_image[:, :, ::-1]
- cropped_faces, restored_faces, gfpgan_output_bgr = gfpgan().enhance(np_image_bgr, has_aligned=False, only_center_face=False, paste_back=True)
- np_image = gfpgan_output_bgr[:, :, ::-1]
-
- return np_image
+ return gfpgan_fix_faces(np_image)
shared.face_restorers.append(FaceRestorerGFPGAN())
except Exception:
|