From b272807bc898a314cde536c1d7d1e43592af1fce Mon Sep 17 00:00:00 2001
From: Joel Schlosser <75754324+jbschlosser@users.noreply.github.com>
Date: Wed, 11 Jun 2025 18:03:40 -0400
Subject: [PATCH] Avoid DtoH sync from access of nonzero() item in scheduler
 (#11696)

---
 src/diffusers/pipelines/flux/pipeline_flux.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/diffusers/pipelines/flux/pipeline_flux.py b/src/diffusers/pipelines/flux/pipeline_flux.py
index 6dbcd5c6db..b9d0b92561 100644
--- a/src/diffusers/pipelines/flux/pipeline_flux.py
+++ b/src/diffusers/pipelines/flux/pipeline_flux.py
@@ -898,6 +898,7 @@ class FluxPipeline(
             )
 
         # 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: