ljh1
/

hello-custom

Text Classification

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

hello-custom / convert_pytorch_to_flax.py

philschmid's picture

philschmid HF staff

Upload convert_pytorch_to_flax.py

8b41d3d about 2 years ago

174 Bytes

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