Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix live preview sigmas with img2img

Browse Source
This commit is contained in:
Disty0
2024-12-27 15:39:58 +03:00
parent ec1f9c89ed
commit d4f04dbd73
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+3 -3
View File
@@ -109,9 +109,9 @@ def diffusers_callback(pipe, step: int = 0, timestep: int = 0, kwargs: dict = {}
shared.state.current_noise_pred = kwargs.get("noise_pred", None)
if shared.state.current_noise_pred is None:
shared.state.current_noise_pred = kwargs.get("predicted_image_embedding", None)
if hasattr(pipe, "scheduler") and hasattr(pipe.scheduler, "sigmas"):
shared.state.current_sigma = pipe.scheduler.sigmas[step]
shared.state.current_sigma_next = pipe.scheduler.sigmas[step + 1]
if hasattr(pipe, "scheduler") and hasattr(pipe.scheduler, "sigmas") and hasattr(pipe.scheduler, "step_index"):
shared.state.current_sigma = pipe.scheduler.sigmas[pipe.scheduler.step_index - 1]
shared.state.current_sigma_next = pipe.scheduler.sigmas[pipe.scheduler.step_index]
except Exception as e:
shared.log.error(f'Callback: {e}')
if shared.cmd_opts.profile and shared.profiler is not None:
modules/shared_state.py
+1 -1
View File
@@ -165,7 +165,7 @@ class State:
self.current_image_sampling_step = self.sampling_step
if self.disable_preview:
return
if self.job == "txt2img" and self.current_noise_pred is not None and self.current_sigma is not None and self.current_sigma_next is not None:
if self.current_noise_pred is not None and self.current_sigma is not None and self.current_sigma_next is not None:
original_sample = sample - (self.current_noise_pred * (self.current_sigma_next-self.current_sigma))
if self.prediction_type in {"epsilon", "flow_prediction"}:
sample = original_sample - (self.current_noise_pred * self.current_sigma)