From 90515c5396834a383fc5e2d4031d3ea51d532a2d Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 3 Sep 2026 08:43:15 -0600 Subject: [PATCH] Fix issue where we gave the wrong dtype hidden states --- extensions_built_in/diffusion_models/ltx2/ltx2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions_built_in/diffusion_models/ltx2/ltx2.py b/extensions_built_in/diffusion_models/ltx2/ltx2.py index ea59521..25770a9 100644 --- a/extensions_built_in/diffusion_models/ltx2/ltx2.py +++ b/extensions_built_in/diffusion_models/ltx2/ltx2.py @@ -994,7 +994,7 @@ class LTX2Model(BaseModel): connector_attention_mask, ) = self.pipeline.connectors( text_embeddings.text_embeds, - text_embeddings.attention_mask.to(self.transformer.dtype), + text_embeddings.attention_mask.to(self.torch_dtype), padding_side=tokenizer_padding_side, ) @@ -1018,7 +1018,7 @@ class LTX2Model(BaseModel): noise_pred_video, noise_pred_audio = self.transformer( hidden_states=packed_latents, - audio_hidden_states=audio_latents.to(self.transformer.dtype), + audio_hidden_states=audio_latents.to(self.torch_dtype), encoder_hidden_states=connector_prompt_embeds, audio_encoder_hidden_states=connector_audio_prompt_embeds, timestep=video_timestep,