Skip to content

HuggingFace:Hub

Download checkpoints

패키지 설치:

pip install huggingface_hub

그리고 다운로드

from huggingface_hub import hf_hub_download

chkpt_path = hf_hub_download("ybelkada/segment-anything", "checkpoints/sam_vit_b_01ec64.pth")

만약 존재하면 다운로드 받지는 않는다. chkpt_path 변수에 문자열로 checkpoint 파일 경로가 반환된다.

See also