RuntimeError: Tensor dimension mismatch in I2V sampling with bf16 model on macOS

[YouNode75]

Issue

When using Wan2_1-I2V-14B-480P_bf16.safetensors model with euler scheduler,
getting tensor dimension mismatch: "The size of tensor a (45) must match the size of tensor b (44) at non-singleton dimension 4"

Environment

• Device: macOS M3 Max (MPS backend)
• Model: wan2.1_i2v_480p_14B_bf16
• Frames: 16, Steps: 4
• Scheduler: euler
• Precision: bf16

Error Location

File: wanvideo/schedulers/fm_solvers_unipc.py, line 511
File: diffusers/schedulers/scheduling_flow_match_euler_discrete.py, line 513

Workaround

Works fine with 1.3B model (Wan2_1-T2V-1_3B_bf16.safetensors)