dtype changed to fp32

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p1.py

@@ -189,7 +189,7 @@ def __call__(
         height=height,
         width=width,
         num_frames=num_frames,
-        dtype=image_embeds.dtype,
+        dtype=jnp.float32,
         rng=latents_rng,
         latents=latents,
         last_image=last_image_tensor,