예제 소스

download_vlm_to_models.py

#!/usr/bin/env python3
"""Qwen3-VL-2B-Instruct 를 OpenAiGym/models/ 아래로 받습니다 (HF hub).

  pip install huggingface_hub
  python download_vlm_to_models.py
"""
from __future__ import annotations

import argparse
from pathlib import Path


def main() -> None:
    p = argparse.ArgumentParser()
    p.add_argument("--repo", default="Qwen/Qwen3-VL-2B-Instruct")
    p.add_argument(
        "--out",
        default=str(Path(__file__).resolve().parent / "models"),
        help="부모 디렉터리 (아래에 Qwen__Qwen3-VL-2B-Instruct 생성)",
    )
    args = p.parse_args()

    from huggingface_hub import snapshot_download

    root = Path(args.out)
    root.mkdir(parents=True, exist_ok=True)
    target = root / args.repo.replace("/", "__")
    print(f"[download] {args.repo}\n[download] -> {target}")
    path = snapshot_download(repo_id=args.repo, local_dir=str(target), max_workers=4)
    print("DONE:", path)


if __name__ == "__main__":
    main()