Fix LTX-2.3 checkpoint config metadata loading
Browse files- _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 |
-
|
| 366 |
-
|
|
|
|
| 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")
|