fix: add size mismatch handling for prev_latents in VideoGenerator to ensure consistent dimensions

lightx2v/models/runners/wan/wan_audio_runner.py

@@ -293,6 +293,15 @@ class VideoGenerator:
         prev_mask[:, prev_frame_len:] = 0
         prev_mask = self._wan_mask_rearrange(prev_mask).unsqueeze(0)
         
+        if prev_latents.shape[-2:] != (height, width):
+            logger.warning(f"Size mismatch: prev_latents {prev_latents.shape} vs scheduler latents (H={height}, W={width}). Config tgt_h={self.config.tgt_h}, tgt_w={self.config.tgt_w}")
+            prev_latents = torch.nn.functional.interpolate(
+                prev_latents,
+                size=(height, width),
+                mode='bilinear',
+                align_corners=False
+            )
+
         return {"prev_latents": prev_latents, "prev_mask": prev_mask}
 
     def _wan_mask_rearrange(self, mask: torch.Tensor) -> torch.Tensor:
@@ -350,6 +359,15 @@ class VideoGenerator:
         prev_mask[:, prev_frame_len:] = 0
         prev_mask = self._wan_mask_rearrange(prev_mask).unsqueeze(0)
         
+        if prev_latents.shape[-2:] != (height, width):
+            logger.warning(f"Size mismatch: prev_latents {prev_latents.shape} vs scheduler latents (H={height}, W={width}). Config tgt_h={self.config.tgt_h}, tgt_w={self.config.tgt_w}")
+            prev_latents = torch.nn.functional.interpolate(
+                prev_latents,
+                size=(height, width),
+                mode='bilinear',
+                align_corners=False
+            )
+
         # Always set previmg_encoder_output
         inputs["previmg_encoder_output"] = {"prev_latents": prev_latents, "prev_mask": prev_mask}