bert-base-uncased-swahili / flax_to_torch.py
fgaim's picture
Add scripts
7b282b4
raw
history blame
131 Bytes
from transformers import BertForMaskedLM
model = BertForMaskedLM.from_pretrained("./", from_flax=True)
model.save_pretrained("./")