Skip to content

Commit 38b0640

Browse files
committed
testing with rope type = interleaved
1 parent b99a37f commit 38b0640

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
141141
ltx2_config = LTX2VideoTransformer3DModel.load_config(config.pretrained_model_name_or_path, subfolder=subfolder)
142142

143143
# Align RoPE type with connectors
144-
ltx2_config["rope_type"] = "split"
144+
ltx2_config["rope_type"] = "interleaved"
145145

146146
if ltx2_config.get("activation_fn") == "gelu-approximate":
147147
ltx2_config["activation_fn"] = "gelu"
@@ -388,7 +388,7 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
388388
"audio_gated_attn": True,
389389
"per_modality_projections": True,
390390
"proj_bias": True,
391-
"rope_type": "split",
391+
"rope_type": "interleaved",
392392
}
393393
)
394394
connector_repo = "Lightricks/LTX-2" if getattr(config, "model_name", "") == "ltx2.3" else config.pretrained_model_name_or_path

0 commit comments

Comments
 (0)