#!/bin/bash

# VENV_PATH="/home/XXXX-3/miniforge3/envs/trainingtt/"

# source "$VENV_PATH/bin/activate"

export WANDB_API_KEY="0054fda6ed25eace6e3c37c9042258f123cccf4c"

source /home/XXXX-11/miniforge3/bin/activate lmfact
module load cuda/12.1
cd LLaMA-Factory/

# python parallel_data_relabel.py --base_save_dir $1 --model_dir $2 --model $3 --batch_size $4
# accelerate launch parallel_data_relabel.py --base_save_dir $1 --model_dir $2 --model $3 --batch_size $4

FORCE_TORCHRUN=1 llamafactory-cli train ../configs/sft_forecasting_qwen25.yaml
conda deactivate