下载模型脚本

执行方法

python download_model.py deepseek-ai/DeepSeek-OCR

输出实例

开始下载模型：deepseek-ai/DeepSeek-OCR
缓存路径：/home/user/vllm/models/
✅ 模型已成功下载到：/home/user/vllm/models/deepseek-ai/DeepSeek-OCR

import sys
from modelscope import snapshot_download

def main():
    # ===== 固定缓存路径变量 =====
    cache_path = "/home/user/vllm/models/"

    # ===== 检查命令行参数 =====
    if len(sys.argv) < 2:
        print("用法: python download_model.py <model_name>")
        print("示例: python download_model.py deepseek-ai/DeepSeek-OCR")
        sys.exit(1)

    model_name = sys.argv[1]

    # ===== 校验输入参数 =====
    if not model_name or '/' not in model_name:
        print(f"❌ 无效的模型名称: {model_name}")
        sys.exit(1)

    try:
        print(f"开始下载模型：{model_name}")
        print(f"缓存路径：{cache_path}")
        model_dir = snapshot_download(model_name, cache_dir=cache_path)
        print(f"✅ 模型已成功下载到：{model_dir}")
    except Exception as e:
        print(f"❌ 模型下载失败: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()

功能说明：

• 你只需要输入模型名称，缓存路径由代码内的变量cache_path 控制。
• 程序仍会验证模型名格式是否正确（例如必须包含/）。
• 出错时自动退出，防止继续执行无效操作。