Skip to content

Commit c7de80b

Browse files
committed
checking why prompt mebeddings is true
1 parent 5e2b98d commit c7de80b

1 file changed

Lines changed: 0 additions & 19 deletions

File tree

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -118,25 +118,6 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
118118
# 1. Load config.
119119
if restored_checkpoint:
120120
ltx2_config = restored_checkpoint["ltx2_config"]
121-
elif getattr(config, "model_name", "") == "ltx2.3":
122-
ltx2_config = {
123-
"in_channels": 128,
124-
"num_attention_heads": 32,
125-
"attention_head_dim": 128,
126-
"cross_attention_dim": 4096,
127-
"audio_in_channels": 128,
128-
"audio_num_attention_heads": 32,
129-
"audio_attention_head_dim": 64,
130-
"audio_cross_attention_dim": 2048,
131-
"num_layers": 48,
132-
"caption_channels": 4096,
133-
"audio_caption_channels": 2048,
134-
"use_prompt_embeddings": True,
135-
"gated_attn": True,
136-
"cross_attn_mod": True,
137-
"audio_gated_attn": True,
138-
"audio_cross_attn_mod": True,
139-
}
140121
else:
141122
ltx2_config = LTX2VideoTransformer3DModel.load_config(config.pretrained_model_name_or_path, subfolder=subfolder)
142123

0 commit comments

Comments
 (0)