download_model.py 239 Bytes
Newer Older
yangzhong's avatar
yangzhong committed
1
2
3
4
5
6
7
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="hfl/chinese-llama-2-7b", local_dir='./pre_model')