File tree Expand file tree Collapse file tree
src/maxdiffusion/pipelines/ltx2 Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -118,25 +118,6 @@ def create_model(rngs: nnx.Rngs, ltx2_config: dict):
118118 # 1. Load config.
119119 if restored_checkpoint :
120120 ltx2_config = restored_checkpoint ["ltx2_config" ]
121- elif getattr (config , "model_name" , "" ) == "ltx2.3" :
122- ltx2_config = {
123- "in_channels" : 128 ,
124- "num_attention_heads" : 32 ,
125- "attention_head_dim" : 128 ,
126- "cross_attention_dim" : 4096 ,
127- "audio_in_channels" : 128 ,
128- "audio_num_attention_heads" : 32 ,
129- "audio_attention_head_dim" : 64 ,
130- "audio_cross_attention_dim" : 2048 ,
131- "num_layers" : 48 ,
132- "caption_channels" : 4096 ,
133- "audio_caption_channels" : 2048 ,
134- "use_prompt_embeddings" : True ,
135- "gated_attn" : True ,
136- "cross_attn_mod" : True ,
137- "audio_gated_attn" : True ,
138- "audio_cross_attn_mod" : True ,
139- }
140121 else :
141122 ltx2_config = LTX2VideoTransformer3DModel .load_config (config .pretrained_model_name_or_path , subfolder = subfolder )
142123
You can’t perform that action at this time.
0 commit comments