From a292d2c47f51fc71cc186709bdf3706f0944b7d6 Mon Sep 17 00:00:00 2001
From: AngelBottomless <aria1th@naver.com>
Date: Wed, 15 Nov 2023 14:26:37 +0900
Subject: [PATCH] hotfix: call shared.state.end() after postprocessing done

---
 modules/postprocessing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/postprocessing.py b/modules/postprocessing.py
index cf04d38b0..fd0c0cc99 100644
--- a/modules/postprocessing.py
+++ b/modules/postprocessing.py
@@ -78,7 +78,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
         image_data.close()
 
     devices.torch_gc()
-
+    shared.state.end()
     return outputs, ui_common.plaintext_to_html(infotext), ''