nb-roberta-base-ncc / convert_model.py
pere's picture
trained
b4c7811
raw
history blame contribute delete
142 Bytes
from transformers import XLMRobertaForMaskedLM
model = XLMRobertaForMaskedLM.from_pretrained(".", from_flax=True)
model.save_pretrained(".")