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

export QUESTIONS_FILE_PATH="./WebQSP/data/WebQSP.test.entities.json"
export QA_WITH_RAG=false

# export LLM_MODEL_TYPE=huggingface
export LLM_MODEL_TYPE=vllm
export HUGGINGFACE_MODEL_NAME_OR_PATH="microsoft/Orca-2-13b"

echo Testing $HUGGINGFACE_MODEL_NAME_OR_PATH
# export RESULT_FILENAME_PREFIX=Tune_2024-04-13_14-36-33
# export REPETITION_PENALTY_START=1.02
python tune_rp.py
# unset RESULT_FILENAME_PREFIX
# unset REPETITION_PENALTY_START

# export HUGGINGFACE_MODEL_NAME_OR_PATH="mistralai/Mistral-7B-Instruct-v0.2"
# echo Testing $HUGGINGFACE_MODEL_NAME_OR_PATH
# python tune_rp.py