Merge pull request #11 from Craftyawesome/fix-gfpgan-vram

Workaround for gfpgan OOM on 6GB

Merge pull request #11 from Craftyawesome/fix-gfpgan-vram
Workaround for gfpgan OOM on 6GB
a51bedfb · AUTOMATIC1111 · GitHub · 21bcbb94 · 92138df5 · a51bedfb
Commit a51bedfb authored Aug 26, 2022 by AUTOMATIC1111 Committed by GitHub Aug 26, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

webui.py webui.py +1 -0

No files found.
--- 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