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