Pārlūkot izejas kodu

call state.jobnext() before postproces*()

Won-Kyu Park 1 gadu atpakaļ
vecāks
revīzija
5121846d34
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  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()
 
+            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))