| # DreamZero 环境快速设置脚本 | |
| # | |
| # 用法: | |
| # source scripts/setup_env.sh # 中国网络（镜像站） | |
| # source scripts/setup_env.sh --offline # 无网络环境 | |
| set -euo pipefail | |
| MODE="${1:-mirror}" | |
| case "$MODE" in | |
| mirror) | |
| echo ">>> 设置 HF 镜像 (hf-mirror.com)" | |
| export HF_ENDPOINT=https://hf-mirror.com | |
| export PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple | |
| export PIP_TRUSTED_HOST=pypi.tuna.tsinghua.edu.cn | |
| ;; | |
| offline) | |
| echo ">>> 离线模式" | |
| export HF_DATASETS_OFFLINE=1 | |
| export TRANSFORMERS_OFFLINE=1 | |
| export HF_HUB_DISABLE_TELEMETRY=1 | |
| ;; | |
| *) | |
| echo "Usage: source scripts/setup_env.sh [mirror|offline]" | |
| return 1 | |
| ;; | |
| esac | |
| echo ">>> 环境配置完成 (MODE=$MODE)" | |