File tree Expand file tree Collapse file tree
src/maxdiffusion/models/ltx2 Expand file tree Collapse file tree Original file line number Diff line number Diff 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 (),
You can’t perform that action at this time.
0 commit comments