from transformers import XLMRobertaForMaskedLM
model = XLMRobertaForMaskedLM.from_pretrained(".",from_flax=True)
model.save_pretrained(".")