File size: 270 Bytes
af38e1d | 1 2 3 4 5 6 | from transformers import TFAutoModelForSequenceClassification
checkpoint_path = "/Users/anjandash/Desktop/HF_MODELS/JavaBERT-small"
tf_model = TFAutoModelForSequenceClassification.from_pretrained(checkpoint_path, from_pt=True)
tf_model.save_pretrained(checkpoint_path)
|