diff --git a/configs/ddim.yaml b/configs/ddim.yaml
index f97578c..b06c4cb 100644
--- a/configs/ddim.yaml
+++ b/configs/ddim.yaml
@@ -4,7 +4,7 @@ fanan:
 
 
 mesh:
-  n_data_parallel: 1
+  n_data_parallel: 4
   n_fsdp_parallel: 1
   n_sequence_parallel: 1
   n_tensors_parallel: 1