NbAiLab
/

roberta_des_512_6e4

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

roberta_des_512_6e4 / generate_pytorch_model.py

pere's picture

first push

5e5a0eb over 2 years ago

raw history blame contribute delete

No virus

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(".")