From b4aa94bc11813190022b14a83eebf55a4912d0fd Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Mon, 23 Feb 2026 19:55:12 +0000 Subject: [PATCH] update --- fastvideo/pipelines/stages/ltx2_denoising.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/fastvideo/pipelines/stages/ltx2_denoising.py b/fastvideo/pipelines/stages/ltx2_denoising.py index 110910f5f..b8c4118cd 100644 --- a/fastvideo/pipelines/stages/ltx2_denoising.py +++ b/fastvideo/pipelines/stages/ltx2_denoising.py @@ -221,7 +221,9 @@ class LTX2DenoisingStage(PipelineStage): stg_scale_audio = batch.ltx2_stg_scale_audio stg_blocks_video = batch.ltx2_stg_blocks_video stg_blocks_audio = batch.ltx2_stg_blocks_audio - do_stg = stg_scale_video != 0.0 or stg_scale_audio != 0.0 + do_stg_video = stg_scale_video != 0.0 + do_stg_audio = stg_scale_audio != 0.0 + do_stg = do_stg_video or do_stg_audio do_cfg_text = (cfg_scale_video != 1.0 or cfg_scale_audio != 1.0) do_mod = (modality_scale_video != 1.0 or modality_scale_audio != 1.0) do_guidance = do_cfg_text or do_mod or do_stg @@ -331,8 +333,10 @@ class LTX2DenoisingStage(PipelineStage): audio_hidden_states=audio_latents, audio_encoder_hidden_states=(audio_context_p), audio_timestep=audio_timestep, - skip_video_self_attn_blocks=(stg_blocks_video), - skip_audio_self_attn_blocks=(stg_blocks_audio), + skip_video_self_attn_blocks=( + stg_blocks_video if do_stg_video else None), + skip_audio_self_attn_blocks=( + stg_blocks_audio if do_stg_audio else None), ) if isinstance(ptb_outputs, tuple): ptb_denoised, ptb_audio = ptb_outputs