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