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)
|