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