File size: 592 Bytes
3236af9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | #!/bin/bash
set -euo pipefail
echo "Requesting one GPU interactive session from Rutgers Slurm..."
echo "When the shell starts on the compute node, run the setup block below."
echo
echo "srun -G 1 --mem=40g -t 04:00:00 --pty /bin/bash -l"
echo
echo "Then inside the session:"
cat <<'EOF'
cd /path/to/NLP_PROJECT_NEW/iconoclast
python3 -m venv .venv
source .venv/bin/activate
python -m pip install --upgrade pip
python -m pip install -e .
export HF_HUB_ENABLE_HF_TRANSFER=1
cp config.llama32_3b.quick.toml config.toml
python -m iconoclast.main \
--model meta-llama/Llama-3.2-3B-Instruct
EOF
|