xinyun/TeaCache
1
0
Fork 0
mirror of https://git.datalinker.icu/ali-vilab/TeaCache synced 2025-12-08 20:34:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix the bug of cnt, https://github.com/ali-vilab/TeaCache/issues/42

Browse Source
This commit is contained in:
Feng Liu 2025-03-07 11:41:49 +08:00 committed by GitHub
parent 1ba2be0a9b
commit 0411d1bd8f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TeaCache4Wan2.1/teacache_generate.py
View File

@ -841,6 +841,7 @@ def generate(args):
# TeaCache # TeaCache
wan_t2v.__class__.generate = t2v_generate wan_t2v.__class__.generate = t2v_generate
wan_t2v.model.__class__.cnt = 0
wan_t2v.model.__class__.enable_teacache = True wan_t2v.model.__class__.enable_teacache = True
wan_t2v.model.__class__.num_steps = args.sample_steps if args.sample_steps is not None else 50 wan_t2v.model.__class__.num_steps = args.sample_steps if args.sample_steps is not None else 50
wan_t2v.model.__class__.rel_l1_thresh = args.teacache_thresh # 2min54s, 0.05: 1min 55s(1.5x), 0.1, 1min 24s(2.1x) 0.15, 1min 6s, 0.08: 1min 27s(2x), 0.07: 1min 48s(1.6x), 0.06: 1min 51s wan_t2v.model.__class__.rel_l1_thresh = args.teacache_thresh # 2min54s, 0.05: 1min 55s(1.5x), 0.1, 1min 24s(2.1x) 0.15, 1min 6s, 0.08: 1min 27s(2x), 0.07: 1min 48s(1.6x), 0.06: 1min 51s
@ -914,6 +915,7 @@ def generate(args):
# TeaCache # TeaCache
wan_i2v.__class__.generate = i2v_generate wan_i2v.__class__.generate = i2v_generate
wan_i2v.model.__class__.cnt = 0
wan_i2v.model.__class__.enable_teacache = True wan_i2v.model.__class__.enable_teacache = True
wan_i2v.model.__class__.num_steps = args.sample_steps if args.sample_steps is not None else 40 wan_i2v.model.__class__.num_steps = args.sample_steps if args.sample_steps is not None else 40
wan_i2v.model.__class__.rel_l1_thresh = args.teacache_thresh # 12min 26s wan_i2v.model.__class__.rel_l1_thresh = args.teacache_thresh # 12min 26s