From e198d833eb31757bfcb2c4784eccfe87d102f521 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 10 Jun 2025 06:15:54 +0000 Subject: [PATCH] enbale teacache and fsdp used both --- generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/generate.py b/generate.py index 739de46..c567440 100644 --- a/generate.py +++ b/generate.py @@ -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