Unverified Commit b272807b authored by Joel Schlosser's avatar Joel Schlosser Committed by GitHub
Browse files

Avoid DtoH sync from access of nonzero() item in scheduler (#11696)

parent 447ccd06
...@@ -898,6 +898,7 @@ class FluxPipeline( ...@@ -898,6 +898,7 @@ class FluxPipeline(
) )
# 6. Denoising loop # 6. Denoising loop
self.scheduler.set_begin_index(0)
with self.progress_bar(total=num_inference_steps) as progress_bar: with self.progress_bar(total=num_inference_steps) as progress_bar:
for i, t in enumerate(timesteps): for i, t in enumerate(timesteps):
if self.interrupt: if self.interrupt:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment