From 387cc23c1a1e6eec65445c4e3dea865d9789d9ae Mon Sep 17 00:00:00 2001 From: Adrian Corduneanu Date: Tue, 25 Feb 2025 23:49:51 -0800 Subject: [PATCH] Update text2video.py only one empty_cache needed before vae decode --- wan/text2video.py | 1 - 1 file changed, 1 deletion(-) diff --git a/wan/text2video.py b/wan/text2video.py index 7e9d4e7..2400545 100644 --- a/wan/text2video.py +++ b/wan/text2video.py @@ -261,7 +261,6 @@ class WanT2V: if offload_model: gc.collect() torch.cuda.synchronize() - torch.cuda.empty_cache() if dist.is_initialized(): dist.barrier()