Peiyuan Zhang
5680039dbe
Update dependencies; Setup code for debugging Delete unused files and code Remove all npu code Update torchvision imports Update EMA model and t2v_debug.sh script Delete npu related stuff and remove inpaint module Remove compress kv Fix warning with dataset handling and model loading Update PyTorch index URLs and video length tolerance range Remove UDIT and inpaint fix typo for dataset download include pretrained open-sora Add pretrained model for OpenSoraT2V-ROPE-L Update max height and width for video processing
19 lines
1.1 KiB
Python
19 lines
1.1 KiB
Python
from huggingface_hub import snapshot_download, hf_hub_download
|
|
import argparse
|
|
# set args for repo_id, local_dir, repo_type,
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Download a dataset or model from the Hugging Face Hub')
|
|
parser.add_argument('--repo_id', type=str, help='The ID of the repository to download')
|
|
parser.add_argument('--local_dir', type=str, help='The local directory to download the repository to')
|
|
parser.add_argument('--repo_type', type=str, help='The type of repository to download (dataset or model)')
|
|
parser.add_argument('--file_name', type=str, help='The file name to download')
|
|
args = parser.parse_args()
|
|
if args.file_name:
|
|
hf_hub_download(repo_id=args.repo_id, filename=args.file_name, repo_type=args.repo_type, local_dir=args.local_dir)
|
|
else:
|
|
snapshot_download(repo_id=args.repo_id,
|
|
local_dir=args.local_dir,
|
|
repo_type=args.repo_type,
|
|
local_dir_use_symlinks=False,
|
|
resume_download=True) |