remove conv_out sharding.

Author: entrpn

src/maxdiffusion/models/wan/transformers/transformer_wan.py

@@ -424,10 +424,6 @@ def __init__(
         dtype=dtype,
         param_dtype=weights_dtype,
         precision=precision,
-        kernel_init=nnx.with_partitioning(
-            nnx.initializers.xavier_uniform(),
-            (None, None, None, None, "conv_out"),
-        ),
     )
 
     # 2. Condition embeddings