Skip to content

Commit e48e104

Browse files
committed
ltx2.3 connectors loading
1 parent ab4841a commit e48e104

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,6 +329,10 @@ def create_model(rngs: nnx.Rngs, config: HyperParameters):
329329
"video_connector_num_layers": 8,
330330
"audio_connector_num_layers": 8,
331331
"caption_channels": 2048,
332+
"video_connector_num_attention_heads": 32,
333+
"audio_connector_num_attention_heads": 32,
334+
"video_connector_attention_head_dim": 64,
335+
"audio_connector_attention_head_dim": 64,
332336
"video_gated_attn": True,
333337
"per_modality_projections": True,
334338
"proj_bias": True,

0 commit comments

Comments
 (0)