Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tqdm logging to the tile loop of TiledStableDiffusionRefineInvocation.

Browse Source
This commit is contained in:
Ryan Dick 2024-07-16 11:28:57 -04:00
parent 4e170baa3a
commit 3027e5bf9d
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/invocations/tiled_stable_diffusion_refine.py
View File

@ -6,6 +6,7 @@ import torch
from diffusers.models.unets.unet_2d_condition import UNet2DConditionModel from diffusers.models.unets.unet_2d_condition import UNet2DConditionModel
from PIL import Image from PIL import Image
from pydantic import field_validator from pydantic import field_validator
from tqdm import tqdm
from invokeai.app.invocations.baseinvocation import BaseInvocation, Classification, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, Classification, invocation
from invokeai.app.invocations.constants import DEFAULT_PRECISION, LATENT_SCALE_FACTOR from invokeai.app.invocations.constants import DEFAULT_PRECISION, LATENT_SCALE_FACTOR
@ -284,8 +285,10 @@ class TiledStableDiffusionRefineInvocation(BaseInvocation):
controlnet_data_tiles.append(tile_controlnet_data) controlnet_data_tiles.append(tile_controlnet_data)
# Denoise (i.e. "refine") each tile independently. # Denoise (i.e. "refine") each tile independently.
for latent_tile, noise_tile, controlnet_data_tile in zip( for latent_tile, noise_tile, controlnet_data_tile in tqdm(
latent_tiles, noise_tiles, controlnet_data_tiles, strict=True zip(latent_tiles, noise_tiles, controlnet_data_tiles, strict=True),
desc="Refining tiles",
total=len(latent_tiles),
): ):
assert latent_tile.shape == noise_tile.shape assert latent_tile.shape == noise_tile.shape