flax-community
/

pino-bigbird-roberta-base

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

pino-bigbird-roberta-base / create_pt_model.py

dat

add pt model

1227ce7 almost 3 years ago

raw history blame

No virus

154 Bytes

	from transformers import BigBirdForMaskedLM
	import torch

	pt_model = BigBirdForMaskedLM.from_pretrained(".", from_flax=True)
	pt_model.save_pretrained(".")