From c2ce4f0a7896ecd75bfbb7a24677e681c424eeb6 Mon Sep 17 00:00:00 2001 From: 16337 <1633794139@qq.com> Date: Mon, 16 Mar 2026 11:31:09 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E4=B8=8B=E8=BD=BD=E8=84=9A=E6=9C=AC=EF=BC=88ModelScope?= =?UTF-8?q?=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Opus 4.6 --- vsp/qwen3.5-9b/download_model.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 vsp/qwen3.5-9b/download_model.py diff --git a/vsp/qwen3.5-9b/download_model.py b/vsp/qwen3.5-9b/download_model.py new file mode 100644 index 0000000..60f2db8 --- /dev/null +++ b/vsp/qwen3.5-9b/download_model.py @@ -0,0 +1,31 @@ +"""从 ModelScope 下载 Qwen3.5-9B 模型""" +import os +import time +import argparse + + +def download_model(model_dir="vsp/qwen3.5-9b/model"): + """下载模型到指定目录""" + from modelscope import snapshot_download + + os.makedirs(model_dir, exist_ok=True) + print(f"开始下载 Qwen3.5-9B 到 {model_dir} ...") + start = time.time() + + model_path = snapshot_download( + "Qwen/Qwen3.5-9B", + cache_dir=model_dir, + ) + + elapsed = time.time() - start + print(f"下载完成!耗时: {elapsed:.1f}s") + print(f"模型路径: {model_path}") + return model_path + + +if __name__ == "__main__": + parser = argparse.ArgumentParser() + parser.add_argument("--model-dir", default="vsp/qwen3.5-9b/model", + help="模型保存目录") + args = parser.parse_args() + download_model(args.model_dir)