philschmid
/

distilbert-base-uncased-emotion

distilbert-base-uncased-emotion / convert_pytorch_to_flax.py

philschmid HF staff

Upload convert_pytorch_to_flax.py

8b41d3d over 1 year ago

No virus

174 Bytes

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