From 3659c3f6e698eda05e4e5346dd77717cd026fb7a Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 28 Jul 2025 10:00:17 +0300
Subject: [PATCH] Progress bar for text encoding

---
 wanvideo/modules/t5.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/wanvideo/modules/t5.py b/wanvideo/modules/t5.py
index 9fc7f67..485d90d 100644
--- a/wanvideo/modules/t5.py
+++ b/wanvideo/modules/t5.py
@@ -2,10 +2,11 @@
 # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
 import logging
 import math
-
+from tqdm import tqdm
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+from comfy.utils import ProgressBar
 
 from .tokenizers import HuggingfaceTokenizer
 
@@ -307,8 +308,10 @@ class T5Encoder(nn.Module):
         x = self.dropout(x)
         e = self.pos_embedding(x.size(1),
                                x.size(1)) if self.shared_pos else None
-        for block in self.blocks:
+        pbar = ProgressBar(len(self.blocks))
+        for block in tqdm(self.blocks, desc="T5Encoder", leave=True):
             x = block(x, mask, pos_bias=e)
+            pbar.update(1)
         x = self.norm(x)
         x = self.dropout(x)
         return x