MickJ commited on
Commit
0746ea8
·
verified ·
1 Parent(s): e0cc94f

Fix LTX-2.3 checkpoint config metadata loading

Browse files
Files changed (1) hide show
  1. _overlay/materialize.py +3 -2
_overlay/materialize.py CHANGED
@@ -362,8 +362,9 @@ def materialize(
362
  )
363
 
364
  source_checkpoint = os.path.join(source_dir, "ltx-2.3-22b-dev.safetensors")
365
- checkpoint_transformer_config = _load_checkpoint_config(source_checkpoint).get(
366
- "transformer", {}
 
367
  )
368
 
369
  transformer_dir = os.path.join(output_dir, "transformer")
 
362
  )
363
 
364
  source_checkpoint = os.path.join(source_dir, "ltx-2.3-22b-dev.safetensors")
365
+ checkpoint_config = _load_checkpoint_config(source_checkpoint)
366
+ checkpoint_transformer_config = checkpoint_config.get(
367
+ "transformer", checkpoint_config
368
  )
369
 
370
  transformer_dir = os.path.join(output_dir, "transformer")