remove conv sharding

Author: entrpn

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

@@ -387,10 +387,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