pino-bigbird-roberta-base / create_pt_model.py
dat
add pt model
1227ce7
raw
history blame
154 Bytes
from transformers import BigBirdForMaskedLM
import torch
pt_model = BigBirdForMaskedLM.from_pretrained(".", from_flax=True)
pt_model.save_pretrained(".")