Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +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/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)