We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e513a2b commit 4c2baffCopy full SHA for 4c2baff
src/maxdiffusion/models/ltx2/text_encoders/text_encoders_ltx2.py
@@ -38,7 +38,7 @@ def __init__(
38
# Feature Extractor Config
39
gemma_dim: int = 3840, # Gemma-3-12b
40
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
+ projection_dim: int = 3840, # LTX-2 conditioning dim
42
# Connector Config
43
connector_heads: int = 32,
44
connector_head_dim: int = 128,
@@ -100,9 +100,9 @@ def __init__(
100
# Feature Extractor Config (Shared)
101
102
103
- projection_dim: int = 4096,
+ projection_dim: int = 3840,
104
105
- connector_heads: int = 32,
+ connector_heads: int = 30,
106
107
connector_layers: int = 2,
108
num_thinking_tokens: int = 128,
0 commit comments