1 files changed, 5 insertions, 6 deletions
diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py
index bb30d733..a5fd9632 100644
--- a/modules/gfpgan_model.py
+++ b/modules/gfpgan_model.py
@@ -44,6 +44,7 @@ def gfpgann():
 
 
 def gfpgan_fix_faces(np_image):
+    global loaded_gfpgan_model
     model = gfpgann()
     if model is None:
         return np_image
@@ -52,7 +53,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
 
@@ -97,11 +100,7 @@ def setup_model(dirname):
                 return "GFPGAN"
 
             def restore(self, np_image):
-                np_image_bgr = np_image[:, :, ::-1]
-                cropped_faces, restored_faces, gfpgan_output_bgr = gfpgann().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: