Skip to content

Commit 753ab0f

Browse files
committed
fix for connectors weight loading
1 parent 118e5d0 commit 753ab0f

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/maxdiffusion/models/ltx2/ltx2_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -406,8 +406,8 @@ def rename_for_ltx2_connector(key):
406406

407407
if "transformer_blocks" in key:
408408
key = key.replace("transformer_blocks", "stacked_blocks")
409-
key = key.replace("ff.net.0.proj", "ff.proj1")
410-
key = key.replace("ff.net.2", "ff.proj2")
409+
key = key.replace("ff.net.0.proj", "ff.net_0")
410+
key = key.replace("ff.net.2", "ff.net_2")
411411
key = key.replace("to_out.0", "to_out")
412412

413413
if key.endswith(".weight"):

0 commit comments

Comments
 (0)