debug

prishajain1 · prishajain1 · commit 71a4e6ab6087 · 2026-05-05T21:51:49.000+05:30
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -1446,6 +1446,7 @@ def _print_stats(name, tensor):
         print(
             f"DEBUG {name} shape: {tensor.shape}, mean: {jnp.round(jnp.mean(tensor), 4)}, min: {jnp.round(jnp.min(tensor), 4)}, max: {jnp.round(jnp.max(tensor), 4)}, std: {jnp.round(jnp.std(tensor), 4)}"
         )
+      print(f"WEIGHT DEBUG: block 0 to_q kernel mean: {self.transformer.transformer_blocks.attn1.to_q.kernel.value[0].mean():.6f}")
       _print_stats("video_embeds", video_embeds)
       _print_stats("audio_embeds", audio_embeds)
 

Original file line number	Diff line number	Diff line change
`@@ -1446,6 +1446,7 @@ def _print_stats(name, tensor):`
`1446`	`1446`	`print(`
`1447`	`1447`	`f"DEBUG {name} shape: {tensor.shape}, mean: {jnp.round(jnp.mean(tensor), 4)}, min: {jnp.round(jnp.min(tensor), 4)}, max: {jnp.round(jnp.max(tensor), 4)}, std: {jnp.round(jnp.std(tensor), 4)}"`
`1448`	`1448`	`)`
	`1449`	`+ print(f"WEIGHT DEBUG: block 0 to_q kernel mean: {self.transformer.transformer_blocks.attn1.to_q.kernel.value[0].mean():.6f}")`
`1449`	`1450`	`_print_stats("video_embeds", video_embeds)`
`1450`	`1451`	`_print_stats("audio_embeds", audio_embeds)`
`1451`	`1452`