Update model.py

2025-03-02 10:52:25 +02:00
parent 55dcc33d00
commit ac98a0318b
1 changed files with 1 additions and 1 deletions
@@ -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(