From e827e2d82a9321e75517d57aa6c09b6aa27c70df Mon Sep 17 00:00:00 2001
From: vbuterin <v@buterin.com>
Date: Thu, 30 Oct 2025 10:12:53 +0300
Subject: [PATCH] Add progress bar to VAE decoding

IMO the VAE decoding step takes long enough that it deserves its own progressbar
---
 wan/modules/vae2_2.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/wan/modules/vae2_2.py b/wan/modules/vae2_2.py
index c0b3f29b..77c49b86 100644
--- a/wan/modules/vae2_2.py
+++ b/wan/modules/vae2_2.py
@@ -6,6 +6,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 from einops import rearrange
+from tqdm import tqdm
 
 __all__ = [
     "Wan2_2_VAE",
@@ -818,7 +819,7 @@ def decode(self, z, scale):
             z = z / scale[1] + scale[0]
         iter_ = z.shape[2]
         x = self.conv2(z)
-        for i in range(iter_):
+        for i in tqdm(range(iter_)):
             self._conv_idx = [0]
             if i == 0:
                 out = self.decoder(