AI

허깅페이스에서 모델 다운받기

반신욕좋아하는J 2024. 9. 10. 09:40
반응형

1. 허깅페이스 웹사이트를 통해 직접 다운로드

  1. 허깅페이스 모델 허브에 접속합니다.
  2. 원하는 모델을 검색하거나 카테고리를 탐색하여 찾습니다.
  3. 모델 페이지에서 "Files and versions" 탭을 클릭합니다.
  4. 다운로드하고 싶은 파일 옆의 다운로드 버튼을 클릭합니다.

2. Transformers 라이브러리 를 통해 다운로드

  1. Transformers 라이브러리 설치
pip install transformers
  1. 모델 다운로드
from transformers import AutoModel, AutoTokenizer

model\_name = "bert-base-uncased"  
model = AutoModel.from\_pretrained(model\_name)  
tokenizer = AutoTokenizer.from\_pretrained(model\_name)

3. CLI 를 통한 다운로드 ( 전 이거씀 )

Git LFS(Large File Storage)를 사용해 다운로드 하는 방법입니다.

  1. Git LFS 설치 (아직 설치하지 않았다면):
    • Linux:
    sudo apt-get install git-lfs
    • macOS:
    brew install git-lfs
  2. Git LFS 초기화:
git lfs install
  1. 모델 저장소 클론
git clone https://huggingface.co/bert-base-uncased
반응형