| #!/usr/bin/env python3 | |
| import logging | |
| from transformers import BertModel, BertTokenizer, TFBertModel | |
| logging.basicConfig(level=logging.INFO, filename="log.txt") | |
| model = BertModel.from_pretrained("../../MiniLM-L12-H384-uncased/") | |
| tf_model = TFBertModel.from_pretrained("../../MiniLM-L12-H384-uncased/", from_pt=True) | |
| model.save_pretrained("./") | |
| tf_model.save_pretrained("./") | |
| tok = BertTokenizer.from_pretrained("../../MiniLM-L12-H384-uncased/") | |
| tok.save_pretrained("./") | |