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

Workaround for gfpgan OOM on 6GB
AUTOMATIC1111 · Aug 26, 2022 · a51bedf · a51bedf
2 parents 21bcbb9 + 92138df
commit a51bedf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/webui.py b/webui.py
@@ -678,6 +678,7 @@ def infotext():
                     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