Progress bar mixin for diffusion pipelines.
Provides a diffusers-compatible progress_bar() method that wraps tqdm, automatically disabling output on non-zero ranks in distributed settings.
Mixin that provides a progress bar for denoising loops.
Usage in pipeline
class MyPipeline(nn.Module, CFGParallelMixin, ProgressBarMixin): def diffuse(self, ...): with self.progress_bar(total=num_steps) as pbar: for i, t in enumerate(timesteps): ... pbar.update()