prevent redundant preview

AUTOMATIC1111 · Dec 6, 2024 · 938ec5e · 938ec5e
1 parent 0120768
commit 938ec5e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
@@ -115,7 +115,11 @@ def images_tensor_to_samples(image, approximation=None, model=None):
 def store_latent(decoded):
     state.current_latent = decoded
 
-    if opts.live_previews_enable and opts.show_progress_every_n_steps > 0 and shared.state.sampling_step % opts.show_progress_every_n_steps == 0:
+    if (
+        (opts.live_previews_enable and opts.show_progress_every_n_steps > 0) and
+        (shared.state.sampling_steps - shared.state.sampling_step > opts.show_progress_every_n_steps) and
+        (shared.state.sampling_step % opts.show_progress_every_n_steps == 0)
+    ):
         if not shared.parallel_processing_allowed:
             shared.state.assign_current_image(sample_to_image(decoded))
 

diff --git a/modules/shared_state.py b/modules/shared_state.py
@@ -139,7 +139,11 @@ def set_current_image(self):
         if not shared.parallel_processing_allowed:
             return
 
-        if self.sampling_step - self.current_image_sampling_step >= shared.opts.show_progress_every_n_steps and shared.opts.live_previews_enable and shared.opts.show_progress_every_n_steps != -1:
+        if (
+            (shared.opts.live_previews_enable and shared.opts.show_progress_every_n_steps != -1) and
+            (self.sampling_steps - self.sampling_step > shared.opts.show_progress_every_n_steps) and
+            (self.sampling_step - self.current_image_sampling_step >= shared.opts.show_progress_every_n_steps)
+        ):
             self.do_set_current_image()
 
     def do_set_current_image(self):