temporarily turn off vae batch parallel when SP is on

mindspore-lab · Sep 27, 2024 · aa45c99 · aa45c99
1 parent c49ae49
commit aa45c99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/opensora_hpcai/scripts/train.py b/examples/opensora_hpcai/scripts/train.py
@@ -377,8 +377,8 @@ def main(args):
             vae = OpenSoraVAE_V1_2(
                 micro_batch_size=args.vae_micro_batch_size,
                 micro_frame_size=args.vae_micro_frame_size,
-                micro_batch_parallel=args.enable_sequence_parallelism,
-                micro_frame_parallel=args.enable_sequence_parallelism,
+                micro_batch_parallel=False,
+                micro_frame_parallel=False,
                 ckpt_path=args.vae_checkpoint,
                 freeze_vae_2d=True,
             )