roberta-swedish / save_torch.py
birgermoell's picture
Added pytorch model
ec35d64
raw
history blame
157 Bytes
from transformers import AutoModelForMaskedLM
model = AutoModelForMaskedLM.from_pretrained(".", from_flax=True)
model.save_pretrained(".", save_config=False)