From 595c827bd31773cc98eb6e87b11090960a32b2a2 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 3 Sep 2022 17:28:30 +0300
Subject: option to unload GFPGAN after using

---
 modules/gfpgan_model.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

(limited to 'modules/gfpgan_model.py')

diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py
index 3f42c163..334a1b7f 100644
--- a/modules/gfpgan_model.py
+++ b/modules/gfpgan_model.py
@@ -4,6 +4,7 @@ import traceback
 
 from modules.paths import script_path
 from modules.shared import cmd_opts
+import modules.shared
 
 
 def gfpgan_model_path():
@@ -23,10 +24,18 @@ loaded_gfpgan_model = None
 def gfpgan():
     global loaded_gfpgan_model
 
-    if loaded_gfpgan_model is None and gfpgan_constructor is not None:
-        loaded_gfpgan_model = gfpgan_constructor(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+    if loaded_gfpgan_model is not None:
+        return loaded_gfpgan_model
 
-    return loaded_gfpgan_model
+    if gfpgan_constructor is None:
+        return None
+
+    model = gfpgan_constructor(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+
+    if not cmd_opts.unload_gfpgan:
+        loaded_gfpgan_model = model
+
+    return model
 
 
 def gfpgan_fix_faces(np_image):
-- 
cgit v1.2.1