File size: 687 Bytes
6d673bc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
#!/bin/sh

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

BASEDIR=`pwd`

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

# Installs Unsloth, Xformers (Flash Attention) and all other packages!
#pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"

pip install unsloth[colab-new] file:///common2/dh.huang.2023/code/unsloth
pip install --no-deps "xformers<0.0.27" "trl<0.9.0" peft accelerate bitsandbytes

export LOGICAL_REASONING_DATA_PATH=datasets/mgtv
export LOGICAL_REASONING_RESULTS_PATH=results/mgtv-results_p2_gemma2.csv
export MODEL_NAME=google/gemma-2-9b-it

python llm_toolkit/tune_logical_reasoning.py