fix

Author: susanbao

run_wan_on_vm.sh

@@ -60,8 +60,8 @@ flash_min_seq_length=0 \
 seed=$RANDOM \
 skip_first_n_steps_for_profiler=3 \
 profiler_steps=3 \
-per_device_batch_size=0.5 \
-ici_data_parallelism=2 \
-ici_fsdp_parallelism=2 \
+per_device_batch_size=0.25 \
+ici_data_parallelism=1 \
+ici_fsdp_parallelism=4 \
 ici_tensor_parallelism=1 \
 enable_ssim=False