debug

prishajain1 · prishajain1 · commit 0f83ffe03ff0 · 2026-02-06T23:08:13.000+05:30
diff --git a/src/maxdiffusion/models/ltx_2/transformer_ltx2.py b/src/maxdiffusion/models/ltx_2/transformer_ltx2.py
@@ -518,7 +518,7 @@ def __init__(
         self.audio_caption_projection = NNXPixArtAlphaTextProjection(
             rngs=rngs, in_features=self.caption_channels, hidden_size=audio_inner_dim, dtype=self.dtype, weights_dtype=self.weights_dtype
         )
-
+        import sys
         print(f"DEBUG IN INIT: inner_dim={inner_dim}, num_attention_heads={num_attention_heads}, attention_head_dim={attention_head_dim}")
         sys.stdout.flush()
         

Original file line number	Diff line number	Diff line change
`@@ -518,7 +518,7 @@ def __init__(`
`518`	`518`	`self.audio_caption_projection = NNXPixArtAlphaTextProjection(`
`519`	`519`	`rngs=rngs, in_features=self.caption_channels, hidden_size=audio_inner_dim, dtype=self.dtype, weights_dtype=self.weights_dtype`
`520`	`520`	`)`
`521`		`-`
	`521`	`+ import sys`
`522`	`522`	`print(f"DEBUG IN INIT: inner_dim={inner_dim}, num_attention_heads={num_attention_heads}, attention_head_dim={attention_head_dim}")`
`523`	`523`	`sys.stdout.flush()`
`524`	`524`