AI
허깅페이스에서 모델 다운받기
반신욕좋아하는J
2024. 9. 10. 09:40
반응형
1. 허깅페이스 웹사이트를 통해 직접 다운로드
- 허깅페이스 모델 허브에 접속합니다.
- 원하는 모델을 검색하거나 카테고리를 탐색하여 찾습니다.
- 모델 페이지에서 "Files and versions" 탭을 클릭합니다.
- 다운로드하고 싶은 파일 옆의 다운로드 버튼을 클릭합니다.
2. Transformers 라이브러리 를 통해 다운로드
- Transformers 라이브러리 설치
pip install transformers
- 모델 다운로드
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)를 사용해 다운로드 하는 방법입니다.
- Git LFS 설치 (아직 설치하지 않았다면):
- Linux:
sudo apt-get install git-lfs
- macOS:
brew install git-lfs
- Windows: Git LFS 웹사이트에서 다운로드
- Git LFS 초기화:
git lfs install
- 모델 저장소 클론
git clone https://huggingface.co/bert-base-uncased
반응형