File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 179
179
"ltx-video" : {"pretrained_model_name_or_path" : "diffusers/LTX-Video-0.9.0" },
180
180
"ltx-video-0.9.1" : {"pretrained_model_name_or_path" : "diffusers/LTX-Video-0.9.1" },
181
181
"ltx-video-0.9.5" : {"pretrained_model_name_or_path" : "Lightricks/LTX-Video-0.9.5" },
182
+ "ltx-video-0.9.7" : {"pretrained_model_name_or_path" : "Lightricks/LTX-Video-0.9.7-dev" },
182
183
"autoencoder-dc-f128c512" : {"pretrained_model_name_or_path" : "mit-han-lab/dc-ae-f128c512-mix-1.0-diffusers" },
183
184
"autoencoder-dc-f64c128" : {"pretrained_model_name_or_path" : "mit-han-lab/dc-ae-f64c128-mix-1.0-diffusers" },
184
185
"autoencoder-dc-f32c32" : {"pretrained_model_name_or_path" : "mit-han-lab/dc-ae-f32c32-mix-1.0-diffusers" },
@@ -644,7 +645,10 @@ def infer_diffusers_model_type(checkpoint):
644
645
model_type = "flux-schnell"
645
646
646
647
elif any (key in checkpoint for key in CHECKPOINT_KEY_NAMES ["ltx-video" ]):
647
- if checkpoint ["vae.encoder.conv_out.conv.weight" ].shape [1 ] == 2048 :
648
+ has_vae = "vae.encoder.conv_in.conv.bias" in checkpoint
649
+ if any (key .endswith ("transformer_blocks.47.scale_shift_table" ) for key in checkpoint ):
650
+ model_type = "ltx-video-0.9.7"
651
+ elif has_vae and checkpoint ["vae.encoder.conv_out.conv.weight" ].shape [1 ] == 2048 :
648
652
model_type = "ltx-video-0.9.5"
649
653
elif "vae.decoder.last_time_embedder.timestep_embedder.linear_1.weight" in checkpoint :
650
654
model_type = "ltx-video-0.9.1"
You can’t perform that action at this time.
0 commit comments