from transformers import BigBirdForMaskedLM
import torch

pt_model = BigBirdForMaskedLM.from_pretrained(".", from_flax=True)
pt_model.save_pretrained(".",save_config=False)