From 7ddfa19bb8cf6660c12161af39730ee688504e29 Mon Sep 17 00:00:00 2001 From: Joel Schlosser Date: Wed, 11 Jun 2025 11:08:04 -0700 Subject: [PATCH] Avoid DtoH sync from access of nonzero() item in scheduler --- 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 6dbcd5c6db54..b9d0b9256109 100644 --- 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: