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

enbale teacache and fsdp used both

Browse Source
This commit is contained in:
Your Name 2025-06-10 06:15:54 +00:00
parent 6686e7fd18
commit e198d833eb
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
generate.py
View File

@ -949,7 +949,8 @@ def generate(args):
if args.enable_teacache:
wan_t2v.__class__.generate = t2v_generate
wan_t2v.model.__class__.enable_teacache = True
wan_t2v.model.__class__.forward = teacache_forward
if args.ulysses_size * args.ring_size == 1: # not conflict with fsdp
wan_t2v.model.__class__.forward = teacache_forward
wan_t2v.model.__class__.cnt = 0
wan_t2v.model.__class__.num_steps = args.sample_steps*2
wan_t2v.model.__class__.teacache_thresh = args.teacache_thresh