roberta-swedish-scandi / save_updated.py
birgermoell's picture
Updated pytorch model export
a27c158
from transformers import AutoTokenizer, RobertaModel
model = RobertaModel.from_pretrained('.',from_flax=True)
tokenizer = AutoTokenizer.from_pretrained(".")
tokenizer.save_pretrained('exported_pytorch_model')
model.save_pretrained('exported_pytorch_model')