fix onediff

pipeline_cogvideox.py

@@ -617,6 +617,7 @@ class CogVideoXPipeline(DiffusionPipeline):
                             **extra_step_kwargs,
                             return_dict=False,
                         )
+                    latents = latents.to(prompt_embeds.dtype)
                     # start diff diff
                     if i < len(timesteps) - 1 and self.original_mask is not None:
                         noise_timestep = timesteps[i + 1]