Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-08 00:04:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update text2video.py only one empty_cache needed before vae decode

Browse Source
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
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
wan/text2video.py
View File

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