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