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)