#!/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 -r requirements.txt
#cd ../LLaMA-Factory && pip install -e .[torch,bitsandbytes] && cd $BASEDIR
#pip install transformers==4.41.2
#pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

export LOGICAL_REASONING_DATA_PATH=datasets/mgtv

export MODEL_NAME=shenzhi-wang/Llama3-8B-Chinese-Chat
# export MODEL_NAME=hfl/llama-3-chinese-8b-instruct-v3

export MODEL_PREFIX=llama3-8b_lora_sft_bf16

export CONFIG_FILE=config/$MODEL_PREFIX-p1_r4.yaml
echo "Tuning $MODEL_NAME with $CONFIG_FILE"
$BASEDIR/scripts/tune-lf.sh $CONFIG_FILE


# export CONFIG_FILE=config/$MODEL_PREFIX-p2_r4.yaml
# echo "Tuning $MODEL_NAME with $CONFIG_FILE"
# $BASEDIR/scripts/tune-lf.sh $CONFIG_FILE


$BASEDIR/scripts/eval-mgtv-llama3_8b.sh