feat: 添加模型下载脚本(ModelScope)

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-03-16 11:31:09 +08:00
parent f29443ffb0
commit c2ce4f0a78

View File

@@ -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)