Spaces:
Build error
Build error
File size: 437 Bytes
a2100ac fc9601b a2100ac e1e71f5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/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
#export YAML=config/mac_template_qwen2_72b.yaml
export PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
python ../llm_toolkit/setup_lf.py
llamafactory-cli train config/models/$MODEL_NAME.yaml
|