blob: 684ded587adf6bf52ada4a2258f965adab885289 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
|
from transformers import BertModel, BertTokenizer
model_name = 'bert-base-uncased'
tokenizer = BertTokenizer.from_pretrained(model_name)
model = BertModel.from_pretrained(model_name)
raw_sentences = ['Tom likes cats', 'Liz likes dogs']
inputs = tokenizer.encode_plus(raw_sentences[0], raw_sentences[1], return_tensors='pt')
# inputs = tokenizer('Hello, my dog is cute', return_tensors='pt')
model(**inputs)
|
