# Single GPU
# python scripts/train_voxcpm_finetune.py --config_path conf/librispeech_full.yaml

# Multi-GPU (4×)
cd /data/mm-llm-backbone_890/personal/sirius/VoxCPM
source ./.venv/bin/activate
source env.sh
source ./.venv/bin/activate
export TORCH_NCCL_HEARTBEAT_TIMEOUT_SEC="${TORCH_NCCL_HEARTBEAT_TIMEOUT_SEC:-3600}"
# print python env info
echo "Python executable: $(which python)"
CUDA_VISIBLE_DEVICES=0,1 torchrun --nproc_per_node=2 \
    scripts/train_voxcpm_finetune.py --config_path conf/TEST_EXP_ffn6144_ditNum4.yaml