machine-translation / scripts /tune-lf-4gpu.sh
dh-mc's picture
Update tune-lf-4gpu.sh
6e8064d
raw
history blame
359 Bytes
#!/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/mac/mac.tsv
export YAML=config/mac_template_4gpu.yaml
python ../llm_toolkit/setup_lf.py
CUDA_VISIBLE_DEVICES=0,1,2,3 llamafactory-cli train config/models/$MODEL_NAME.yaml