[LTXPipeline] Update latents dtype to match VAE dtype (#11533)

james-p-xu · web-flow · commit 3c0a0129fe08 · 2025-05-09T16:05:21.000+05:30
fix: update latents dtype to match vae
diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx.py b/src/diffusers/pipelines/ltx/pipeline_ltx.py
@@ -789,6 +789,7 @@ def __call__(
                 ]
                 latents = (1 - decode_noise_scale) * latents + decode_noise_scale * noise
 
+            latents = latents.to(self.vae.dtype)
             video = self.vae.decode(latents, timestep, return_dict=False)[0]
             video = self.video_processor.postprocess_video(video, output_type=output_type)
 

Original file line number	Diff line number	Diff line change
`@@ -789,6 +789,7 @@ def __call__(`
`789`	`789`	`]`
`790`	`790`	`latents = (1 - decode_noise_scale) * latents + decode_noise_scale * noise`
`791`	`791`
	`792`	`+ latents = latents.to(self.vae.dtype)`
`792`	`793`	`video = self.vae.decode(latents, timestep, return_dict=False)[0]`
`793`	`794`	`video = self.video_processor.postprocess_video(video, output_type=output_type)`
`794`	`795`