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

source "$(cd "$(dirname "$0")" && pwd)/_common.sh"
require_venv

# nanoGPT README lists these deps.
"${VENV_PY}" -m pip install --upgrade pip
"${VENV_PY}" -m pip install torch numpy transformers datasets tiktoken wandb tqdm

# Quick sanity check
"${VENV_PY}" -c "import torch; print('torch', torch.__version__); print('mps', torch.backends.mps.is_available())"
