[INFO|parser.py:352] 2024-08-31 17:53:54,714 >> Process rank: 0, device: mps, n_gpu: 1, distributed training: False, compute dtype: torch.bfloat16