LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2025-12-11 15:04:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
fix: update latents dtype to match vae
This commit is contained in:
James Xu
2025-05-09 03:35:21 -07:00
committed by GitHub
parent 2d380895e5
commit 3c0a0129fe
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/pipelines/ltx/pipeline_ltx.py
View File

@@ -789,6 +789,7 @@ class LTXPipeline(DiffusionPipeline, FromSingleFileMixin, LTXVideoLoraLoaderMixi
]
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)