Skip to content

Commit 7e8870c

Browse files
committed
use_prompt_embeddings set to true
1 parent 97f5dc2 commit 7e8870c

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
@@ -131,7 +131,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
131131
"num_layers": 48,
132132
"caption_channels": 3840,
133133
"audio_caption_channels": 2048,
134-
"use_prompt_embeddings": False,
134+
"use_prompt_embeddings": True,
135135
"gated_attn": False,
136136
"cross_attn_mod": True,
137137
"audio_gated_attn": False,
@@ -162,7 +162,7 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
162162
ltx2_config["gated_attn"] = False
163163
ltx2_config["cross_attn_mod"] = True
164164
ltx2_config["perturbed_attn"] = True
165-
ltx2_config["use_prompt_embeddings"] = False
165+
ltx2_config["use_prompt_embeddings"] = True
166166

167167
# 2. eval_shape
168168
p_model_factory = partial(create_model, ltx2_config=ltx2_config)

0 commit comments

Comments
 (0)