flax-community
/

roberta-swahili

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

roberta-swahili / flax_to_torch.py

fgaim's picture

set new dataset in train_tokenizer

e5bd3ab over 3 years ago

137 Bytes

	from transformers import RobertaForMaskedLM

	model = RobertaForMaskedLM.from_pretrained("./", from_flax=True)
	model.save_pretrained("./")