Skip to content

Commit db87a47

Browse files
committed
feat(ltx2): remove unused text_encoder_kernel from specs
1 parent 0216b16 commit db87a47

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

src/maxdiffusion/models/ltx2/logical_sharding_ltx2.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,6 @@ class TextEncoderShardingSpecs:
5757
"""Specs for the Text Encoder execution."""
5858

5959
use_batched_text_encoder: bool = False # Flag to batch text encoder execution
60-
text_encoder_kernel: Optional[tuple] = None # Spec for text encoder projection layers
6160

6261

6362
@dataclass
@@ -103,7 +102,6 @@ class VAEShardingSpecs:
103102
),
104103
"text_encoder": TextEncoderShardingSpecs(
105104
use_batched_text_encoder=True,
106-
text_encoder_kernel=(None, "embed"),
107105
),
108106
"text_connector": TextConnectorShardingSpecs(
109107
qkv_kernel=(None, "heads"),
@@ -116,7 +114,6 @@ class VAEShardingSpecs:
116114
"ltx2_dit": LTX2DiTShardingSpecs(),
117115
"text_encoder": TextEncoderShardingSpecs(
118116
use_batched_text_encoder=False,
119-
text_encoder_kernel=(None, "embed"),
120117
),
121118
"text_connector": TextConnectorShardingSpecs(),
122119
"vae": VAEShardingSpecs(),

0 commit comments

Comments
 (0)