amending the runscripts to use the new installation

This commit is contained in:
Ruhollah Majdoddin 2025-10-30 19:36:22 +01:00
parent bdeacffdae
commit f35b27c4a2
No known key found for this signature in database
GPG Key ID: A28D46DCC3E05D49
2 changed files with 4 additions and 6 deletions

View File

@ -12,6 +12,8 @@
export OMP_NUM_THREADS=1 export OMP_NUM_THREADS=1
export NANOCHAT_BASE_DIR="$HOME/.cache/nanochat" export NANOCHAT_BASE_DIR="$HOME/.cache/nanochat"
mkdir -p $NANOCHAT_BASE_DIR mkdir -p $NANOCHAT_BASE_DIR
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
source "$HOME/.cargo/env"
command -v uv &> /dev/null || curl -LsSf https://astral.sh/uv/install.sh | sh command -v uv &> /dev/null || curl -LsSf https://astral.sh/uv/install.sh | sh
[ -d ".venv" ] || uv venv [ -d ".venv" ] || uv venv
uv sync --extra cpu uv sync --extra cpu
@ -19,9 +21,6 @@ source .venv/bin/activate
if [ -z "$WANDB_RUN" ]; then if [ -z "$WANDB_RUN" ]; then
WANDB_RUN=dummy WANDB_RUN=dummy
fi fi
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
source "$HOME/.cargo/env"
uv run maturin develop --release --manifest-path rustbpe/Cargo.toml
EVAL_BUNDLE_URL=https://karpathy-public.s3.us-west-2.amazonaws.com/eval_bundle.zip EVAL_BUNDLE_URL=https://karpathy-public.s3.us-west-2.amazonaws.com/eval_bundle.zip
if [ ! -d "$NANOCHAT_BASE_DIR/eval_bundle" ]; then if [ ! -d "$NANOCHAT_BASE_DIR/eval_bundle" ]; then
curl -L -o eval_bundle.zip $EVAL_BUNDLE_URL curl -L -o eval_bundle.zip $EVAL_BUNDLE_URL

View File

@ -8,6 +8,8 @@
export OMP_NUM_THREADS=1 export OMP_NUM_THREADS=1
export NANOCHAT_BASE_DIR="$HOME/.cache/nanochat" export NANOCHAT_BASE_DIR="$HOME/.cache/nanochat"
mkdir -p $NANOCHAT_BASE_DIR mkdir -p $NANOCHAT_BASE_DIR
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
source "$HOME/.cargo/env"
command -v uv &> /dev/null || curl -LsSf https://astral.sh/uv/install.sh | sh command -v uv &> /dev/null || curl -LsSf https://astral.sh/uv/install.sh | sh
[ -d ".venv" ] || uv venv [ -d ".venv" ] || uv venv
uv sync --extra gpu uv sync --extra gpu
@ -16,9 +18,6 @@ if [ -z "$WANDB_RUN" ]; then
WANDB_RUN=dummy WANDB_RUN=dummy
fi fi
python -m nanochat.report reset python -m nanochat.report reset
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
source "$HOME/.cargo/env"
uv run maturin develop --release --manifest-path rustbpe/Cargo.toml
EVAL_BUNDLE_URL=https://karpathy-public.s3.us-west-2.amazonaws.com/eval_bundle.zip EVAL_BUNDLE_URL=https://karpathy-public.s3.us-west-2.amazonaws.com/eval_bundle.zip
if [ ! -d "$NANOCHAT_BASE_DIR/eval_bundle" ]; then if [ ! -d "$NANOCHAT_BASE_DIR/eval_bundle" ]; then
curl -L -o eval_bundle.zip $EVAL_BUNDLE_URL curl -L -o eval_bundle.zip $EVAL_BUNDLE_URL