hotfix: call shared.state.end() after postprocessing done

a292d2c4 · AngelBottomless · GitHub · 5e80d9ee · a292d2c4
Commit a292d2c4 authored Nov 15, 2023 by AngelBottomless Committed by GitHub Nov 15, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/postprocessing.py modules/postprocessing.py +1 -1

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