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(".") |
1227ce7 |
1 2 3 4 5 |
from transformers import BigBirdForMaskedLM
import torch
pt_model = BigBirdForMaskedLM.from_pretrained(".", from_flax=True)
pt_model.save_pretrained(".") |