From de304b10d124266ba85cfea744becdc52d3eb0ff Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 19 Sep 2025 12:45:08 +0300
Subject: [PATCH] face adapter offloading

---
 wanvideo/modules/model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 790cbc8..1150ddf 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -1906,7 +1906,9 @@ class WanModel(torch.nn.Module):
     def wananimate_forward(self, block_idx, x, motion_vec, strength=1.0, motion_masks=None):
         if block_idx % 5 == 0:
             adapter_args = [x, motion_vec, motion_masks]
+            self.face_adapter.to(self.main_device)
             residual_out = self.face_adapter.fuser_blocks[block_idx // 5](*adapter_args)
+            self.face_adapter.to(self.offload_device)
             return x.add(residual_out, alpha=strength)
         return x