#!/bin/sh

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

BASEDIR=`pwd`

export LOGICAL_REASONING_DATA_PATH=datasets/mgtv
export RESIZE_TOKEN_EMBEDDINGS=true
#export USING_LLAMA_FACTORY=true
export USING_P1_PROMPT_TEMPLATE=false
export LOAD_IN_4BIT=true

export ORG_NAME=$1
export MODEL=$2
export MODEL_NAME=$ORG_NAME/$MODEL

export LOGICAL_REASONING_RESULTS_PATH=data/${MODEL}_results.csv

echo Evaluating $MODEL_NAME with few-shot learning
python llm_toolkit/eval_shots.py

if [ "$MODEL" == "Llama3.1-70B-Chinese-Chat" ];
then
    export START_NUM_SHOTS=5
    $BASEDIR/scripts/eval-shots.sh internlm internlm2_5-20b-chat
fi