From 6e063c12fbfbac0805398f2bfa8c6df053cd962f Mon Sep 17 00:00:00 2001 From: FlyingBlackShark Date: Mon, 18 May 2026 04:42:27 +0000 Subject: [PATCH 1/2] Fix long sequence NaN issue --- src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py index 70ea97c4..d9fd20c5 100644 --- a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py +++ b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py @@ -1347,7 +1347,7 @@ def __call__( video_sequence_length *= (height // self.vae_spatial_compression_ratio) * (width // self.vae_spatial_compression_ratio) mu = calculate_shift( - video_sequence_length, + 4096, self.scheduler.config.get("base_image_seq_len", 1024), self.scheduler.config.get("max_image_seq_len", 4096), self.scheduler.config.get("base_shift", 0.95), From 3a498aea0325b33946c5c213cc89977a1075e1fd Mon Sep 17 00:00:00 2001 From: FlyingBlackShark Date: Thu, 21 May 2026 16:01:15 +0000 Subject: [PATCH 2/2] Cap LTX2 dynamic shift sequence length --- src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py index d9fd20c5..edc2af5d 100644 --- a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py +++ b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py @@ -1345,9 +1345,11 @@ def __call__( video_sequence_length = (num_frames - 1) // self.vae_temporal_compression_ratio + 1 video_sequence_length *= (height // self.vae_spatial_compression_ratio) * (width // self.vae_spatial_compression_ratio) + # Cap long videos while preserving smaller videos' dynamic shift behavior. + shift_sequence_length = min(6144, video_sequence_length) mu = calculate_shift( - 4096, + shift_sequence_length, self.scheduler.config.get("base_image_seq_len", 1024), self.scheduler.config.get("max_image_seq_len", 4096), self.scheduler.config.get("base_shift", 0.95),