from transformers import RobertaForMaskedLM, AutoTokenizer model = RobertaForMaskedLM.from_pretrained("./", from_flax=True) model.save_pretrained("./") tokenizer = AutoTokenizer.from_pretrained("./") tokenizer.save_pretrained("./")