[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:
1 parent
0f9ee6afd8
commit
ea14b04e9f
12 files changed
+76
-55
No files matched your search
@@ -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)")
|
||||
|
||||
Reference in new issue
Block a user