예제 소스

download_qwen3vl.py

#!/usr/bin/env python3
"""Qwen3-VL-2B-Instruct 가중치/프로세서 파일을 vlm_models/ 로 다운로드.

사용:
  python download_qwen3vl.py
  python download_qwen3vl.py --repo Qwen/Qwen3-VL-2B-Instruct
"""
from __future__ import annotations

import argparse
import os
from pathlib import Path


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("--repo", default="Qwen/Qwen3-VL-2B-Instruct")
    parser.add_argument(
        "--out",
        default=str(Path(__file__).resolve().parent.parent / "vlm_models"),
        help="모델을 저장할 부모 디렉토리 (그 아래 <repo_id 슬래시→__> 폴더 생성).",
    )
    args = parser.parse_args()

    from huggingface_hub import snapshot_download

    target = Path(args.out) / args.repo.replace("/", "__")
    target.mkdir(parents=True, exist_ok=True)
    print(f"[download] repo={args.repo}\n[download] dir ={target}")

    path = snapshot_download(
        repo_id=args.repo,
        local_dir=str(target),
        max_workers=4,
    )
    print(f"DONE → {path}")
    print("로컬 경로로 서버 실행 예시:")
    print(f"  python vlm_pva_server.py --host 127.0.0.1 --port 9299 --device cpu \\\n    --model-id {path!r}")


if __name__ == "__main__":
    main()