From f2b7c13ffe4a43aa8d4dd5e758d5ad563b71a55f Mon Sep 17 00:00:00 2001 From: Franc1sCai Date: Tue, 27 Jan 2026 09:52:13 +0800 Subject: [PATCH] [BugFix]: wan22 animate block logic bug --- .../networks/wan/weights/animate/transformer_weights.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lightx2v/models/networks/wan/weights/animate/transformer_weights.py b/lightx2v/models/networks/wan/weights/animate/transformer_weights.py index 57fcdfb0..18c9ceea 100755 --- a/lightx2v/models/networks/wan/weights/animate/transformer_weights.py +++ b/lightx2v/models/networks/wan/weights/animate/transformer_weights.py @@ -29,8 +29,8 @@ def _add_animate_fuserblock_to_offload_buffers(self): if hasattr(self, "offload_block_cuda_buffers") and self.offload_block_cuda_buffers is not None: for i in range(self.offload_blocks_num): self.offload_block_cuda_buffers[i].compute_phases.append(WanAnimateFuserBlock(self.config, 0, "face_adapter.fuser_blocks", self.mm_type, create_cuda_buffer=True)) - if self.lazy_load: - self.offload_block_cpu_buffers[i].compute_phases.append(WanAnimateFuserBlock(self.config, 0, "face_adapter.fuser_blocks", self.mm_type, create_cpu_buffer=True)) + if self.lazy_load: + self.offload_block_cpu_buffers[i].compute_phases.append(WanAnimateFuserBlock(self.config, 0, "face_adapter.fuser_blocks", self.mm_type, create_cpu_buffer=True)) elif hasattr(self, "offload_phase_cuda_buffers") and self.offload_phase_cuda_buffers is not None: self.offload_phase_cuda_buffers.append(WanAnimateFuserBlock(self.config, 0, "face_adapter.fuser_blocks", self.mm_type, create_cuda_buffer=True)) if self.lazy_load: