scripts/tune-lf_v2_4bit.sh

#!/bin/sh

BASEDIR=$(dirname "$0")
cd $BASEDIR/../llama-factory/
echo Current Directory:
pwd

export ORG_NAME=$1
export MODEL_NAME=$2
export CHAT_TEMPLATE=$3
export DATA_PATH=../datasets/mgtv
export YAML=config/mgtv_template_4bit.yaml

export PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"

python ../llm_toolkit/setup_lf.py
llamafactory-cli train config/models/$MODEL_NAME.yaml