File size: 154 Bytes
1227ce7
 
 
 
 
1
2
3
4
5
from transformers import BigBirdForMaskedLM
import torch

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