Skip to content

Commit 4c2baff

Browse files
committed
modify config val
Signed-off-by: James Huang <syhuang1201@gmail.com>
1 parent e513a2b commit 4c2baff

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/maxdiffusion/models/ltx2/text_encoders/text_encoders_ltx2.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(
3838
# Feature Extractor Config
3939
gemma_dim: int = 3840, # Gemma-3-12b
4040
gemma_layers: int = 49, # Gemma-3 has 48 layers + 1 embedding layer output = 49 hidden states
41-
projection_dim: int = 4096, # LTX-2 conditioning dim
41+
projection_dim: int = 3840, # LTX-2 conditioning dim
4242
# Connector Config
4343
connector_heads: int = 32,
4444
connector_head_dim: int = 128,
@@ -100,9 +100,9 @@ def __init__(
100100
# Feature Extractor Config (Shared)
101101
gemma_dim: int = 3840, # Gemma-3-12b
102102
gemma_layers: int = 49, # Gemma-3 has 48 layers + 1 embedding layer output = 49 hidden states
103-
projection_dim: int = 4096,
103+
projection_dim: int = 3840,
104104
# Connector Config
105-
connector_heads: int = 32,
105+
connector_heads: int = 30,
106106
connector_head_dim: int = 128,
107107
connector_layers: int = 2,
108108
num_thinking_tokens: int = 128,

0 commit comments

Comments
 (0)