summaryrefslogtreecommitdiffstatshomepage
path: root/main.py
blob: de058e750540a521b4093c38c3c354c2758f6ab9 (plain) (blame)
1
2
3
4
from transformers import BertTokenizer, BertModel

tokenizer = BertTokenizer.from_pretrained("hfl/minirbt-h256")
model = BertModel.from_pretrained("hfl/minirbt-h256")