"...text-generation-inference.git" did not exist on "53c2c3dbc7cb8a7d91d519a898fc8d74eae91e48"
Unverified Commit b272807b authored by Joel Schlosser's avatar Joel Schlosser Committed by GitHub
Browse files

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

parent 447ccd06
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment