feat: 添加模型下载脚本(ModelScope)
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
31
vsp/qwen3.5-9b/download_model.py
Normal file
31
vsp/qwen3.5-9b/download_model.py
Normal 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)
|
||||
Reference in New Issue
Block a user