debug

prishajain1 · prishajain1 · commit f2cbf5a240bf · 2026-02-06T22:55:41.000+05:30
diff --git a/src/maxdiffusion/models/ltx_2/transformer_ltx2.py b/src/maxdiffusion/models/ltx_2/transformer_ltx2.py
@@ -507,6 +507,8 @@ def __init__(
             rngs=rngs, in_features=self.caption_channels, hidden_size=audio_inner_dim, dtype=self.dtype, weights_dtype=self.weights_dtype
         )
 
+        print(f"DEBUG IN INIT: inner_dim={inner_dim}, num_attention_heads={num_attention_heads}, attention_head_dim={attention_head_dim}")
+        
         # 3. Timestep Modulation Params and Embedding
         self.time_embed = LTX2AdaLayerNormSingle(
             rngs=rngs, embedding_dim=inner_dim, num_mod_params=6, use_additional_conditions=False, dtype=self.dtype, weights_dtype=self.weights_dtype

Original file line number	Diff line number	Diff line change
`@@ -507,6 +507,8 @@ def __init__(`
`507`	`507`	`rngs=rngs, in_features=self.caption_channels, hidden_size=audio_inner_dim, dtype=self.dtype, weights_dtype=self.weights_dtype`
`508`	`508`	`)`
`509`	`509`
	`510`	`+ print(f"DEBUG IN INIT: inner_dim={inner_dim}, num_attention_heads={num_attention_heads}, attention_head_dim={attention_head_dim}")`
	`511`	`+`
`510`	`512`	`# 3. Timestep Modulation Params and Embedding`
`511`	`513`	`self.time_embed = LTX2AdaLayerNormSingle(`
`512`	`514`	`rngs=rngs, embedding_dim=inner_dim, num_mod_params=6, use_additional_conditions=False, dtype=self.dtype, weights_dtype=self.weights_dtype`