diff --git a/src/diffusers/pipelines/wan/pipeline_wan_animate.py b/src/diffusers/pipelines/wan/pipeline_wan_animate.py
index a923219a7550..c2416243f111 100644
--- a/src/diffusers/pipelines/wan/pipeline_wan_animate.py
+++ b/src/diffusers/pipelines/wan/pipeline_wan_animate.py
@@ -465,8 +465,10 @@ def get_i2v_mask(
         mask_len: int = 1,
         mask_pixel_values: torch.Tensor | None = None,
         dtype: torch.dtype | None = None,
-        device: str | torch.device = "cuda",
+        device: str | torch.device | None = None,
     ) -> torch.Tensor:
+        device = device or self._execution_device
+
         # mask_pixel_values shape (if supplied): [B, C = 1, T, latent_h, latent_w]
         if mask_pixel_values is None:
             mask_lat_size = torch.zeros(