1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

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

This commit is contained in:
Joel Schlosser
2025-06-11 18:03:40 -04:00
committed by GitHub
parent 447ccd0679
commit b272807bc8

View File

@@ -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: