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" "$@"