From ac98a0318bd8bb820340ebf33d47f2b05801f626 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 2 Mar 2025 10:52:25 +0200
Subject: [PATCH] Update model.py

---
 wanvideo/modules/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 73f28c2..f264f6b 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -566,7 +566,7 @@ class WanModel(ModelMixin, ConfigMixin):
         log.info(f"Block swap memory summary:")
         log.info(f"Transformer blocks on {self.offload_device}: {total_offload_memory:.2f}MB")
         log.info(f"Transformer blocks on {self.main_device}: {total_main_memory:.2f}MB")
-        log.info(f"Total Memory: {(total_offload_memory + total_main_memory):.2f}MB")
+        log.info(f"Total memory used by transformer blocks: {(total_offload_memory + total_main_memory):.2f}MB")
         log.info("----------------------")
 
     def forward(