yi_lu/ti_coding_agent_probe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Hugging Face model download CLI

Browse Source
This commit is contained in:
Codex
2026-06-25 15:34:22 +08:00
parent e79c7ea9bf
commit 4f9e0a30a9
3 changed files with 44 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
scripts/download_models.sh
View File

@@ -15,6 +15,26 @@ 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}"
huggingface-cli download "${QWEN35_9B_MODEL_ID}" --local-dir "models/qwen3.5-9b" --local-dir-use-symlinks False
huggingface-cli download "${QWEN36_27B_MODEL_ID}" --local-dir "models/qwen3.6-27b" --local-dir-use-symlinks False
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"