Jelajahi Sumber

Merge pull request #13762 from wkpark/nextjob

call state.jobnext() before postproces*()
AUTOMATIC1111 1 tahun lalu
induk
melakukan
21d561885e
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  1. 2 2
      modules/processing.py

+ 2 - 2
modules/processing.py

@@ -886,6 +886,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
 
 
             devices.torch_gc()
             devices.torch_gc()
 
 
+            state.nextjob()
+
             if p.scripts is not None:
             if p.scripts is not None:
                 p.scripts.postprocess_batch(p, x_samples_ddim, batch_number=n)
                 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()
             devices.torch_gc()
 
 
-            state.nextjob()
-
         if not infotexts:
         if not infotexts:
             infotexts.append(Processed(p, []).infotext(p, 0))
             infotexts.append(Processed(p, []).infotext(p, 0))