File size: 798 Bytes
fb11af9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import argparse
import os
from huggingface_hub import snapshot_download
"""
python3 scripts/download_hf_model.py --repo_id deepseek-ai/Janus-1.3B --local_dir Janus-1.3B
"""
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--repo_id", type=str, default="deepseek-ai/Janus-1.3B")
parser.add_argument("--local_dir", type=str, default="./Janus-1.3B")
parser.add_argument("--local_dir_use_symlinks", type=bool, default=False)
args = parser.parse_args()
repo_id = args.repo_id
local_dir = args.local_dir
local_dir_use_symlinks = args.local_dir_use_symlinks
snapshot_download(
repo_id=repo_id,
local_dir=os.path.join(local_dir, repo_id.split("/")[1]),
local_dir_use_symlinks=local_dir_use_symlinks,
)
|