1#!/usr/bin/env bash
 2
 3set -e
 4
 5QUANTIZED_MODEL="${1:-"$QUANTIZED_MODEL"}"
 6BUILD_DIR="${2:-"$BUILD_DIR"}"
 7
 8if [ -z "$QUANTIZED_MODEL" ]; then
 9    echo "Error: Model path must be provided either as:" >&2
10    echo "  1. Command line argument" >&2
11    echo "  2. QUANTIZED_MODEL environment variable" >&2
12    exit 1
13fi
14
15# Check if data/wikitext-2-raw directory exists
16if [ ! -d "ppl/wikitext-2-raw" ]; then
17    echo "ppl/wikitext-2-raw directory does not exist. Downloading..." >&2
18    mkdir -p ppl
19    pushd ppl
20    ./../../../scripts/get-wikitext-2.sh
21    popd
22fi
23
24if [ -z "$BUILD_DIR" ]; then
25    BUILD_DIR="../../build"
26fi
27
28cmake --build $BUILD_DIR --target llama-perplexity -j8
29
30${BUILD_DIR}/bin/llama-perplexity -m $QUANTIZED_MODEL -f ppl/wikitext-2-raw/wiki.test.raw
31
32