| import argparse | |
| from huggingface_hub import snapshot_download | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument('--folder_path', type=str) | |
| parser.add_argument('--repo_id', type=str) | |
| parser.add_argument('--read_token', type=str) | |
| "doesn't save to huggingface cache, only downloads in specified local directory" | |
| if __name__ == "__main__": | |
| args = parser.parse_args() | |
| snapshot_download( | |
| repo_id=args.repo_id, | |
| repo_type="model", | |
| local_dir=args.folder_path, | |
| local_dir_use_symlinks=False, | |
| token=args.read_token | |
| ) |