41 lines
1.1 KiB
Bash
Executable File
41 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
cd "${ROOT_DIR}"
|
|
|
|
export http_proxy="${http_proxy:-http://100.72.0.101:8888}"
|
|
export https_proxy="${https_proxy:-http://100.72.0.101:8888}"
|
|
export HTTP_PROXY="${HTTP_PROXY:-${http_proxy}}"
|
|
export HTTPS_PROXY="${HTTPS_PROXY:-${https_proxy}}"
|
|
export HF_ENDPOINT="${HF_ENDPOINT:-https://hf-mirror.com}"
|
|
|
|
source .venv/bin/activate
|
|
mkdir -p models
|
|
|
|
QWEN35_9B_MODEL_ID="${QWEN35_9B_MODEL_ID:-Qwen/Qwen3.5-9B}"
|
|
QWEN36_27B_MODEL_ID="${QWEN36_27B_MODEL_ID:-Qwen/Qwen3.6-27B}"
|
|
HF_DOWNLOAD_MAX_WORKERS="${HF_DOWNLOAD_MAX_WORKERS:-8}"
|
|
|
|
download_model() {
|
|
local model_id="$1"
|
|
local local_dir="$2"
|
|
local cmd=(
|
|
hf download "${model_id}"
|
|
--local-dir "${local_dir}"
|
|
--max-workers "${HF_DOWNLOAD_MAX_WORKERS}"
|
|
)
|
|
if [[ -n "${HF_TOKEN:-}" ]]; then
|
|
cmd+=(--token "${HF_TOKEN}")
|
|
fi
|
|
printf '%q ' "${cmd[@]}"
|
|
echo
|
|
if [[ "${DRY_RUN:-0}" == "1" ]]; then
|
|
return 0
|
|
fi
|
|
"${cmd[@]}"
|
|
}
|
|
|
|
download_model "${QWEN35_9B_MODEL_ID}" "models/qwen3.5-9b"
|
|
download_model "${QWEN36_27B_MODEL_ID}" "models/qwen3.6-27b"
|