TILDE / README.md
ielab's picture
Update README.md
de04dcb
|
raw
history blame
277 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')

Github: https://github.com/ielab/TILDE