We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7847b19 commit 403e710Copy full SHA for 403e710
1 file changed
src/maxdiffusion/models/ltx2/ltx2_utils.py
@@ -336,7 +336,7 @@ def load_transformer_weights(
336
for k in list(random_flax_state_dict.keys()):
337
k_str = str(k)
338
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
+ print(f"EVAL_SHAPE: {k}")
340
pass
341
342
for pt_key, tensor in tensors.items():
0 commit comments