Ver código fonte

call state.jobnext() before postproces*()

Won-Kyu Park 1 ano atrás
pai
commit
5121846d34
1 arquivos alterados com 2 adições e 2 exclusões
  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))