roberta-swedish-scandi / save_torch.py
birgermoell's picture
Added ppytorch model
f98be5f
raw
history blame
136 Bytes
from transformers import RobertaForMaskedLM
model = RobertaForMaskedLM.from_pretrained("./", from_flax=True)
model.save_pretrained("./")