roberta_jan_128_scandinavian / generate_pytorch_model.py

first submit of new scandinavian model based on roberta_jan_128_ncc

ad93372 about 3 years ago

256 Bytes

	# This script overwrites any existing PyTorch model. Generates a new one with an LM head from the pretrained Flax model.
	from transformers import RobertaForMaskedLM
	model = RobertaForMaskedLM.from_pretrained(".",from_flax=True)
	model.save_pretrained(".")