Pre-training python src/main_pretrain.py --base_config src/config/train_server/released_v0/base_180m_wo_obelics.yaml --variant_config src/config/train_server/released_v0/2b_200m_image_only_optimal.yaml --deepspeed src/config/deepspeed/deepspeed_config.json