From b8f8a08ba40e9a7a53db6a9a36f489f5be68f983 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Wed, 15 Jul 2026 12:17:06 -0600 Subject: [PATCH] Fix sampling bar with anima --- extensions_built_in/diffusion_models/anima/anima.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/extensions_built_in/diffusion_models/anima/anima.py b/extensions_built_in/diffusion_models/anima/anima.py index 0fd483f..9d3772b 100644 --- a/extensions_built_in/diffusion_models/anima/anima.py +++ b/extensions_built_in/diffusion_models/anima/anima.py @@ -341,6 +341,18 @@ class AnimaModel(BaseModel): vae=unwrap_model(self.vae), ) pipeline = pipeline.to(self.device_torch) + + # ModularPipeline.set_progress_bar_config only walks one level of sub_blocks, + # but the tqdm bar lives in the loop block nested two levels deep. Must use + # _blocks; the public .blocks property returns a fresh copy on every access. + def disable_progress_bars(blocks): + for sub_block in blocks.sub_blocks.values(): + if hasattr(sub_block, "set_progress_bar_config"): + sub_block.set_progress_bar_config(disable=True) + if hasattr(sub_block, "sub_blocks"): + disable_progress_bars(sub_block) + + disable_progress_bars(pipeline._blocks) return pipeline def _offload_text_encoder(self):