changing embed to None

Author: prishajain1

src/maxdiffusion/configs/ltx2_video.yml

@@ -47,9 +47,9 @@ logical_axis_rules: [
                       ['activation_length', 'context'],
                       ['activation_heads', 'tensor'],
                       ['mlp','tensor'],
-                      ['embed', ['context', 'fsdp']],
+                      ['embed', None],
                       ['heads', 'tensor'],
-                      ['norm', ['fsdp']],
+                      ['norm', 'tensor'],
                       ['conv_batch', ['data', 'context', 'fsdp']],
                       ['out_channels', 'tensor'],
                       ['conv_out', 'context'],