Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion modules/sd_samplers_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))

Expand Down
6 changes: 5 additions & 1 deletion modules/shared_state.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
Loading