Save BERT locally, by the name of the model

Save BERT locally, by the name of the BERT model

from transformers import AutoTokenizer, AutoModel
savepppath = "/Users/xxxx/Desktop/info/BERT/"
tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/bert-base-nli-mean-tokens")
model = AutoModel.from_pretrained("sentence-transformers/bert-base-nli-mean-tokens")
tokenizer.save_pretrained(savepppath)
model.save_pretrained(savepppath)

Leave a Reply

Your email address will not be published. Required fields are marked *

Proudly powered by WordPress | Theme: Rits Blog by Crimson Themes.