fixes to smoke tes

Author: amepas

src/maxdiffusion/configs/base_flux2klein.yml

@@ -159,8 +159,8 @@ dcn_data_parallelism: 1  # recommended DCN axis to be auto-sharded
 dcn_fsdp_parallelism: -1
 dcn_context_parallelism: 1
 dcn_tensor_parallelism: 1
-ici_data_parallelism: -1
-ici_fsdp_parallelism: 1  # recommended ICI axis to be auto-sharded
+ici_data_parallelism: 1
+ici_fsdp_parallelism: -1  # recommended ICI axis to be auto-sharded
 ici_context_parallelism: 1
 ici_tensor_parallelism: 1