Method init_cache

diffsynth/models/wan_video_dit.py:410–413

ctx: [B, S_ctx, dim] —— 经过 text_embedding 之后的上下文

(self, ctx: torch.Tensor)

    @torch.no_grad()
    def init_cache(self, ctx: torch.Tensor):
        """ctx: [B, S_ctx, dim] —— 经过 text_embedding 之后的上下文"""
        self.cache_k = self.norm_k(self.k(ctx))
        self.cache_v = self.v(ctx)

    def clear_cache(self):
        self.cache_k = None