from transformers import BertForSequenceClassification | |
model = BertForSequenceClassification.from_pretrained("./", from_flax=True) | |
model.save_pretrained("./") | |
from transformers import BertForSequenceClassification | |
model = BertForSequenceClassification.from_pretrained("./", from_flax=True) | |
model.save_pretrained("./") | |