Skip to content

Commit

Permalink
update printing message
Browse files Browse the repository at this point in the history
  • Loading branch information
wtomin committed Sep 17, 2024
1 parent e00d3c8 commit f8a7a1e
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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}",
Expand Down

0 comments on commit f8a7a1e

Please sign in to comment.