File size: 787 Bytes
6d1c39a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25aea89
6d1c39a
 
 
25aea89
6d1c39a
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#!/bin/sh

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

nvidia-smi
uname -a
cat /etc/os-release
lscpu
grep MemTotal /proc/meminfo

# pip install -r requirements.txt
# FLASH_ATTENTION_FORCE_BUILD=TRUE pip install --upgrade flash-attn

export MODEL_NAME=unsloth/Qwen2-0.5B-Instruct
echo Tuning $MODEL_NAME
python llm_toolkit/tune_mac.py

export MODEL_NAME=unsloth/Qwen2-1.5B-Instruct
echo Tuning $MODEL_NAME
python llm_toolkit/tune_mac.py

export MODEL_NAME=unsloth/Qwen2-7B-Instruct
echo Tuning $MODEL_NAME
python llm_toolkit/tune_mac.py

export MODEL_NAME=unsloth/mistral-7b-instruct-v0.3
echo Tuning $MODEL_NAME
python llm_toolkit/tune_mac.py

export MODEL_NAME=gradientai/Llama-3-8B-Instruct-Gradient-1048k
echo Tuning $MODEL_NAME
python llm_toolkit/tune_mac.py