Skip to content

Commit 403e710

Browse files
committed
fix
1 parent 7847b19 commit 403e710

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/models/ltx2/ltx2_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ def load_transformer_weights(
336336
for k in list(random_flax_state_dict.keys()):
337337
k_str = str(k)
338338
if "transformer_blocks" in k_str and ("attn1" in k_str or "ff" in k_str):
339-
# print(f"EVAL_SHAPE: {k}") # Comment out to reduce noise, we know they exist
339+
print(f"EVAL_SHAPE: {k}")
340340
pass
341341

342342
for pt_key, tensor in tensors.items():

0 commit comments

Comments
 (0)