Cleanup memory after model run

2024-08-30 20:32:17 +00:00 · 2023-09-01 20:50:39 +03:00 · 2023-09-01 20:50:39 +03:00 · 5151798a16
commit 5151798a16
parent 1a9f552a75
1 changed files with 1 additions and 0 deletions
--- a/invokeai/backend/image_util/lama.py
+++ b/invokeai/backend/image_util/lama.py
@ -53,5 +53,6 @@ class LaMA:

        del model
        gc.collect()
+        torch.cuda.empty_cache()

        return infilled_image