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(