bhadresh-savani
/

roberta-base-emotion

Text Classification

Inference Endpoints

Model card Files Files and versions Community

roberta-base-emotion / convert_pytorch_to_flax.py

bhadresh-savani's picture

bhadresh-savani

added flax and tensorflow model

90a5f86 over 2 years ago

raw history blame

No virus

174 Bytes

	from transformers import FlaxAutoModelForSequenceClassification
	model = FlaxAutoModelForSequenceClassification.from_pretrained("./", from_pt=True)
	model.save_pretrained("./")