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 progress bar to FLUX denoising.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-20 14:52:05 +00:00 committed by Brandon Rising
parent 5e2351f3bf
commit 120e1cf1e9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/flux/sampling.py
View File

@ -4,6 +4,7 @@ from typing import Callable
import torch import torch
from einops import rearrange, repeat from einops import rearrange, repeat
from torch import Tensor from torch import Tensor
from tqdm import tqdm
from .model import Flux from .model import Flux
from .modules.conditioner import HFEncoder from .modules.conditioner import HFEncoder
@ -115,7 +116,7 @@ def denoise(
# this is ignored for schnell # this is ignored for schnell
guidance_vec = torch.full((img.shape[0],), guidance, device=img.device, dtype=img.dtype) guidance_vec = torch.full((img.shape[0],), guidance, device=img.device, dtype=img.dtype)
for t_curr, t_prev in zip(timesteps[:-1], timesteps[1:], strict=True): for t_curr, t_prev in tqdm(list(zip(timesteps[:-1], timesteps[1:], strict=True))):
t_vec = torch.full((img.shape[0],), t_curr, dtype=img.dtype, device=img.device) t_vec = torch.full((img.shape[0],), t_curr, dtype=img.dtype, device=img.device)
pred = model( pred = model(
img=img, img=img,