Skip to content

Commit fc0c4db

Browse files
committed
dim mismatch fix
1 parent a0c2fdc commit fc0c4db

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/models/wan/wan_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def load_base_wan_transformer(
261261
# This ensures the transpose happens regardless of what rename_key did.
262262
if "net.0" in renamed_pt_key or "net_0" in renamed_pt_key or \
263263
"net.2" in renamed_pt_key or "net_2" in renamed_pt_key:
264-
tensor = tensor.T()
264+
tensor = tensor.T
265265

266266
# 2. FIX net_0: Strip .proj (Handle both raw and renamed keys)
267267
if "net.0.proj" in renamed_pt_key:

0 commit comments

Comments
 (0)