From d6864ec46e2b06735f40f318ef7e067885c3f858 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 7 Oct 2025 23:48:26 +0300
Subject: [PATCH] Fix for S2V

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

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 0db73fd..c2d777b 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -2572,7 +2572,10 @@ class WanModel(torch.nn.Module):
                     should_calc = True
 
         x = x.to(self.base_dtype)
-        e0 = e0.to(self.base_dtype)
+        if isinstance(e0, list):
+            e0 = [item.to(self.base_dtype) if torch.is_tensor(item) else item for item in e0]
+        else:
+            e0 = e0.to(self.base_dtype)
 
         if self.enable_easycache:
             original_x = x.clone().to(self.cache_device)