Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +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: