remove conv sharding.

entrpn · entrpn · commit 8040f29a116a · 2025-08-14T23:30:33.000Z
diff --git a/src/maxdiffusion/configs/base_wan_14b.yml b/src/maxdiffusion/configs/base_wan_14b.yml
@@ -140,7 +140,6 @@ logical_axis_rules: [
                       ['norm', 'tensor'],
                       ['conv_batch', ['data','fsdp']],
                       ['out_channels', 'tensor'],
-                      ['conv_out', 'fsdp'],
                     ]
 data_sharding: [['data', 'fsdp', 'tensor']]
 

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,6 @@ logical_axis_rules: [`
`140`	`140`	`['norm', 'tensor'],`
`141`	`141`	`['conv_batch', ['data','fsdp']],`
`142`	`142`	`['out_channels', 'tensor'],`
`143`		`- ['conv_out', 'fsdp'],`
`144`	`143`	`]`
`145`	`144`	`data_sharding: [['data', 'fsdp', 'tensor']]`
`146`	`145`