CUDA_VISISBLE_DEVICES=0,1,2,3,4,5 accelerate launch --config_file='configs/accelerator_config.yaml' src.train.py --report_to='wandb'  --checkpointing_steps="best" --save_every=5 --config='tangoflux_config.yaml'