kobert 설치하기

 

 

 

 

kobert 설치하기

 

KoBERT는 SK텔레콤에서 공개한 한국어 BERT 모델로, PyTorch 및 TensorFlow에서 사용할 수 있습니다. 일반적으로 kobert-transformers 또는 kobert 패키지를 설치해서 사용합니다. 

 

✅ 1. 기본 환경 준비 (Python 3.6~3.10 권장)

pip install torch torchvision

 

※ python 3.13 까지 설치 됩니다. 

 

✅ 2. kobert 설치 (huggingface 호환 버전)

 

pip install kobert-transformers

 

또는 원본에 더 가까운 monologg/kobert 버전 설치:

pip install git+https://github.com/SKTBrain/KoBERT.git@master

 

---

✅ 3. 예제 코드

from kobert_transformers

import get_tokenizer from transformers

import BertModel # 토크나이저와 모델 로드

tokenizer = get_tokenizer()

model = BertModel.from_pretrained("monologg/kobert") # 인풋 토큰화

text = "안녕하세요. KoBERT를 설치했습니다."

inputs = tokenizer(text, return_tensors="pt") # 모델 추론

outputs = model(**inputs)

print(outputs.last_hidden_state.shape)

---

✅ 추가 팁

• CUDA 환경에서 사용하려면 torch 버전에 GPU용 설치가 필요합니다.
• transformers 라이브러리도 함께 설치되는 경우가 많으며, 직접 지정하고 싶으면:

pip install transformers==4.30.2