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

jbschlosser · web-flow · commit b272807bc898 · 2025-06-11T12:03:40.000-10:00
diff --git a/src/diffusers/pipelines/flux/pipeline_flux.py b/src/diffusers/pipelines/flux/pipeline_flux.py
@@ -898,6 +898,7 @@ def __call__(
             )
 
         # 6. Denoising loop
+        self.scheduler.set_begin_index(0)
         with self.progress_bar(total=num_inference_steps) as progress_bar:
             for i, t in enumerate(timesteps):
                 if self.interrupt:

Original file line number	Diff line number	Diff line change
`@@ -898,6 +898,7 @@ def __call__(`
`898`	`898`	`)`
`899`	`899`
`900`	`900`	`# 6. Denoising loop`
	`901`	`+ self.scheduler.set_begin_index(0)`
`901`	`902`	`with self.progress_bar(total=num_inference_steps) as progress_bar:`
`902`	`903`	`for i, t in enumerate(timesteps):`
`903`	`904`	`if self.interrupt:`