roberta-debug-8 / generate_pt_model.py
pere's picture
first submit
a4a3924
from transformers import XLMRobertaForMaskedLM, XLMRobertaConfig
config = XLMRobertaConfig.from_pretrained("./")
model = XLMRobertaForMaskedLM.from_pretrained("./",config=config,from_flax=True)
model.save_pretrained("./")