From f8a7a1e3981f1a5c0a837adbb9259c5b296a4120 Mon Sep 17 00:00:00 2001 From: Didan Deng <33117903+wtomin@users.noreply.github.com> Date: Tue, 17 Sep 2024 13:50:22 +0800 Subject: [PATCH] update printing message --- examples/opensora_pku/opensora/train/train_t2v_diffusers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/opensora_pku/opensora/train/train_t2v_diffusers.py b/examples/opensora_pku/opensora/train/train_t2v_diffusers.py index 426ae00df7..609f21cf30 100644 --- a/examples/opensora_pku/opensora/train/train_t2v_diffusers.py +++ b/examples/opensora_pku/opensora/train/train_t2v_diffusers.py @@ -554,7 +554,7 @@ def main(args): f"Jit level: {args.jit_level}", f"Distributed mode: {args.use_parallel}" + (f"\nParallel mode: {args.parallel_mode}" if args.use_parallel else "") - + (f"\nsp_size {args.sp_size}" if args.sp_size != 1 else ""), + + (f"\nsp_size: {args.sp_size}" if args.sp_size != 1 else ""), f"Num params: {num_params:,} (transformer: {num_params_transformer:,}, vae: {num_params_vae:,})", f"Num trainable params: {num_params_trainable:,}", f"Transformer model dtype: {model_dtype}",