From aa45c99acdb0eca93ac112dac757fffb54e7258a Mon Sep 17 00:00:00 2001 From: Mike Cheung Date: Fri, 27 Sep 2024 13:44:39 +0800 Subject: [PATCH] temporarily turn off vae batch parallel when SP is on --- examples/opensora_hpcai/scripts/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/opensora_hpcai/scripts/train.py b/examples/opensora_hpcai/scripts/train.py index cbbcd70e34..6a0b4f2cb1 100644 --- 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, )