File size: 503 Bytes
fc30f26
 
13a5c22
fc30f26
 
493a5f1
 
 
fc30f26
 
 
 
 
 
 
 
 
13a5c22
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import os
from huggingface_hub import snapshot_download
from transformers.utils.hub import move_cache

hf_token: str = os.getenv("HF_TOKEN")
if hf_token is None:
    raise ValueError("HF_TOKEN is not set")

hf_token = hf_token.strip()
if hf_token == "":
    raise ValueError("HF_TOKEN is empty")

snapshot_download(
    repo_id="sail/Sailor-4B-Chat",
    revision="89a866a7041e6ec023dd462adeca8e28dd53c83e",
    token=hf_token,
)

# https://github.com/huggingface/transformers/issues/20428
move_cache()