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

Live previews add flow_prediction

Browse Source
This commit is contained in:
Disty0
2024-12-25 18:43:10 +03:00
parent 5e35d6c3a3
commit 917d9e6c86
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+1 -1
View File
@@ -119,7 +119,7 @@ def diffusers_callback(pipe, step: int = 0, timestep: int = 0, kwargs: dict = {}
if "flow" in pipe.scheduler.__class__.__name__.lower():
shared.state.current_latent = original_sample - (noise_pred * sigma)
elif hasattr(pipe.scheduler, "config") and hasattr(pipe.scheduler.config, "prediction_type"):
if pipe.scheduler.config.prediction_type == "epsilon":
if pipe.scheduler.config.prediction_type in {"epsilon", "flow_prediction"}:
shared.state.current_latent = original_sample - (noise_pred * sigma)
elif pipe.scheduler.config.prediction_type == "v_prediction":
shared.state.current_latent = noise_pred * (-sigma / (sigma**2 + 1) ** 0.5) + (original_sample / (sigma**2 + 1))