Skip to content

Commit f09f51c

Browse files
committed
undo config mapping fn changes
1 parent 428db9f commit f09f51c

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/diffusers/loaders/single_file_model.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,11 +86,9 @@
8686
},
8787
"LTXTransformer3DModel": {
8888
"checkpoint_mapping_fn": convert_ltx_transformer_checkpoint_to_diffusers,
89-
"config_mapping_fn": lambda *_, **__: {},
9089
},
9190
"AutoencoderKLLTX": {
9291
"checkpoint_mapping_fn": convert_ltx_vae_checkpoint_to_diffusers,
93-
"config_mapping_fn": lambda *_, **__: {},
9492
},
9593
}
9694

@@ -229,7 +227,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
229227
mapping_functions = SINGLE_FILE_LOADABLE_CLASSES[mapping_class_name]
230228

231229
checkpoint_mapping_fn = mapping_functions["checkpoint_mapping_fn"]
232-
if original_config is not None:
230+
if original_config:
233231
if "config_mapping_fn" in mapping_functions:
234232
config_mapping_fn = mapping_functions["config_mapping_fn"]
235233
else:

0 commit comments

Comments
 (0)