distilbert-base-uncased-sentiment-sst2 / convert_flax_to_pytorch.py

added pytorch and tensorflow model

ba72176 over 2 years ago

No virus

168 Bytes

	from transformers import AutoModelForSequenceClassification
	model = AutoModelForSequenceClassification.from_pretrained("./", from_flax=True)
	model.save_pretrained("./")