#!/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

# 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 LOAD_IN_4BIT=false
export USING_LLAMA_FACTORY=false
export START_EPOCH=0

export MODEL_NAME=google/gemma-2-9b-it
export USING_P1_PROMPT_TEMPLATE=false
export LOGICAL_REASONING_DATA_PATH=datasets/mgtv

export MODEL_PREFIX=gemma2-9b_lora_sft_bf16
export LOGICAL_REASONING_RESULTS_PATH=results/$MODEL_PREFIX-p2.csv
export ADAPTER_PATH_BASE=unsloth/gemma-2-9b-it

echo "Eval $MODEL_NAME with $ADAPTER_PATH_BASE"
python llm_toolkit/eval_logical_reasoning_all_epochs.py