TILDE / README.md
ielab's picture
Create README.md
f8864d2
|
raw
history blame
237 Bytes

Please treat TILDE as a BertLMHeadModel model:

from transformers import BertLMHeadModel, BertTokenizerFast

model = BertLMHeadModel.from_pretrained("ielab/TILDE")
tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')