File size: 435 Bytes
3738140 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | #!/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" "$@"
|