aboutsummaryrefslogtreecommitdiff
path: root/webui.py
diff options
context:
space:
mode:
Diffstat (limited to 'webui.py')
-rw-r--r--webui.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/webui.py b/webui.py
index 2751d75a..b3375e98 100644
--- a/webui.py
+++ b/webui.py
@@ -678,6 +678,7 @@ def process_images(outpath, func_init, func_sample, prompt, seed, sampler_index,
x_sample = x_sample.astype(np.uint8)
if use_GFPGAN and GFPGAN is not None:
+ torch_gc()
cropped_faces, restored_faces, restored_img = GFPGAN.enhance(x_sample, has_aligned=False, only_center_face=False, paste_back=True)
x_sample = restored_img