From 380e9a84c3e8ca37d5ad6f39d4aef75545fd80ca Mon Sep 17 00:00:00 2001
From: Won-Kyu Park <wkpark@gmail.com>
Date: Thu, 19 Sep 2024 10:24:59 +0900
Subject: [PATCH] call lowvram.send_everything_to_cpu() for interrupted case

---
 modules/processing.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index 3b23ab7af..ed140983d 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -1147,6 +1147,11 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
     if p.scripts is not None:
         p.scripts.postprocess(p, res)
 
+
+    if lowvram.is_enabled(shared.sd_model):
+        # for interrupted case
+        lowvram.send_everything_to_cpu()
+
     return res