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