github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix sampling bar with anima

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-15 12:17:06 -06:00
parent 3e6bd874c4
commit b8f8a08ba4
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
extensions_built_in/diffusion_models/anima/anima.py
View File

@@ -341,6 +341,18 @@ class AnimaModel(BaseModel):
vae=unwrap_model(self.vae), vae=unwrap_model(self.vae),
) )
pipeline = pipeline.to(self.device_torch) 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 return pipeline
def _offload_text_encoder(self): def _offload_text_encoder(self):