CODELOCK-8Bbeta / scripts /setup_cloud.sh
CODELOCKHF's picture
Release v0.2: multichannel affect engine and calibrated vectors
031e289
Raw History Blame Contribute Delete
805 Bytes
#!/usr/bin/env bash
# Linux/RunPod: isolated Python environment reusing the image's CUDA torch.
set -euo pipefail
cd "$(dirname "$0")/.."
python3 -m venv --system-site-packages .venv-v2
PYTHON=.venv-v2/bin/python
"$PYTHON" -m pip install -e '.[dev]'
nvidia-smi --query-gpu=name,memory.total --format=csv
"$PYTHON" - <<'PY'
from pathlib import Path
from huggingface_hub import snapshot_download
target = Path('models/Qwen3-8B')
index = target / 'model.safetensors.index.json'
if not index.is_file():
snapshot_download('Qwen/Qwen3-8B', local_dir=target,
allow_patterns=['*.json', '*.safetensors', '*.txt', '*.model', 'tokenizer*'])
print('Base model:', target.resolve())
PY
"$PYTHON" -m pytest
echo 'Ready. Run: .venv-v2/bin/python scripts/run_pipeline.py --model models/Qwen3-8B'