| 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, | |
| ) | |