#!/bin/bash
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export TORCH_NCCL_AVOID_RECORD_STREAMS=1
export HF_ENDPOINT="https://hf-mirror.com"
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7

python -m accelerate.commands.launch \
    --config_file ./path/to/config.yaml \
    --num_processes 8 \
    script/train/SFT_HR2R.py \
    --config ./path/to/config.yaml