tucan9389 / pytorch-Korean-NER

Apache License 2.0
6 stars 0 forks source link

kobert 대신 kcbert 사용 #4

Closed tucan9389 closed 3 years ago

tucan9389 commented 3 years ago
tucan9389 commented 3 years ago

KcBERT

from transformers import AutoTokenizer, AutoModelWithLMHead

# Base Model (108M)

tokenizer = AutoTokenizer.from_pretrained("beomi/kcbert-base")
model = AutoModelWithLMHead.from_pretrained("beomi/kcbert-base")

# Large Model (334M)

tokenizer = AutoTokenizer.from_pretrained("beomi/kcbert-large")
model = AutoModelWithLMHead.from_pretrained("beomi/kcbert-large")
tucan9389 commented 3 years ago

KcELECTRA

from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("beomi/KcELECTRA-base")
model = AutoModel.from_pretrained("beomi/KcELECTRA-base")