Update text2video.py only one empty_cache needed before vae decode

This commit is contained in:
Adrian Corduneanu 2025-02-25 23:49:51 -08:00 committed by GitHub
parent 65819c1d08
commit 387cc23c1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -261,7 +261,6 @@ class WanT2V:
if offload_model:
gc.collect()
torch.cuda.synchronize()
torch.cuda.empty_cache()
if dist.is_initialized():
dist.barrier()