from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained(".", from_flax=True)
model.save_pretrained(".", save_config=False)