down_model.py 232 Bytes
Newer Older
yangzhong's avatar
yangzhong committed
1
2
3
4
5
import os
os.environ['CURL_CA_BUNDLE'] = ''
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
from huggingface_hub import hf_hub_download, snapshot_download
snapshot_download(repo_id="SherryX/STAR", local_dir='pretrained_weight')