call state.jobnext() before postproces*()

5121846d · Won-Kyu Park · 464fbcd9 · 5121846d
Commit 5121846d authored Oct 25, 2023 by Won-Kyu Park
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

modules/processing.py modules/processing.py +2 -2

No files found.
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -886,6 +886,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:

            devices.torch_gc()

+            state.nextjob()
+
            if p.scripts is not None:
                p.scripts.postprocess_batch(p, x_samples_ddim, batch_number=n)

@@ -958,8 +960,6 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:

            devices.torch_gc()

-            state.nextjob()
-
        if not infotexts:
            infotexts.append(Processed(p, []).infotext(p, 0))