Skip to content

Commit 3bdad89

Browse files
committed
[DtoH sync] set scheduler begin index in SD3 T2I
Reference: #11696
1 parent 0196914 commit 3bdad89

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1058,6 +1058,12 @@ def __call__(
10581058
self._joint_attention_kwargs.update(ip_adapter_image_embeds=ip_adapter_image_embeds)
10591059

10601060
# 7. Denoising loop
1061+
1062+
# We set the index here to remove DtoH sync, helpful especially during compilation.
1063+
# Check out more details here: https://github.com/huggingface/diffusers/pull/11696
1064+
if hasattr(self.scheduler, "set_begin_index"):
1065+
self.scheduler.set_begin_index(0)
1066+
10611067
with self.progress_bar(total=num_inference_steps) as progress_bar:
10621068
for i, t in enumerate(timesteps):
10631069
if self.interrupt:

0 commit comments

Comments
 (0)