mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-06-07 07:44:53 +00:00
Update text2video.py only one empty_cache needed before vae decode
This commit is contained in:
parent
65819c1d08
commit
387cc23c1a
@ -261,7 +261,6 @@ class WanT2V:
|
||||
if offload_model:
|
||||
gc.collect()
|
||||
torch.cuda.synchronize()
|
||||
torch.cuda.empty_cache()
|
||||
if dist.is_initialized():
|
||||
dist.barrier()
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user