[Bugfix][BAGEL] Restore original denoise step count (#7049)

Signed-off-by: Trigger <Meng.Bo.Wang@outlook.com>
Signed-off-by: Gao Han <hgaoaf@connect.ust.hk>
Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
This commit is contained in:
TriggerandGao Han authored and GitHub committed 2026-09-04 17:05:05 +00:00
1 parent 0f9ee6afd8
commit ea14b04e9f
12 files changed
+76 -55

No files matched your search

+5 -1
View File
@@ -1141,7 +1141,11 @@ class DiffusionEngine:
height=512,
width=512,
guidance_scale=0.0,
num_inference_steps=1,
# Dummy warmup must exercise at least one denoising iteration in
# every execution mode. Some pipelines (for example BAGEL) perform
# ``num_inference_steps - 1`` scheduler updates and reject an empty
# one-step schedule.
num_inference_steps=2,
)
if req is None:
logger.info("Skipping dummy warmup run (num_frames=0)")