Skip to content

GitLab

Unverified Commit 3c0a0129 authored by James Xu's avatar James Xu Committed by GitHub
Browse files

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

fix: update latents dtype to match vae
parent 2d380895
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/pipelines/ltx/pipeline_ltx.py
View file @ 3c0a0129
...@@ -789,6 +789,7 @@ class LTXPipeline(DiffusionPipeline, FromSingleFileMixin, LTXVideoLoraLoaderMixi ...@@ -789,6 +789,7 @@ class LTXPipeline(DiffusionPipeline, FromSingleFileMixin, LTXVideoLoraLoaderMixi
] ]
latents = (1 - decode_noise_scale) * latents + decode_noise_scale * noise 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.vae.decode(latents, timestep, return_dict=False)[0]
video = self.video_processor.postprocess_video(video, output_type=output_type) video = self.video_processor.postprocess_video(video, output_type=output_type)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment