#!/usr/bin/env bash
set -euo pipefail

repo_root="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"

export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"
export MODEL="${MODEL:-meta-llama/Llama-2-7b-hf}"
export OUTPUT_ROOT="${OUTPUT_ROOT:-$repo_root/results/uidl_llama}"
export SIM_DATASET="${SIM_DATASET:-slimpajama}"
export SIM_DATASET_CONFIG="${SIM_DATASET_CONFIG:-none}"

"$repo_root/compare_model/UIDL/run_uidl_prune.sh" "$@"