| #!/usr/bin/env python | |
| from transformers import AutoTokenizer, TFAutoModelForCausalLM | |
| import tensorflow as tf | |
| tokenizer = AutoTokenizer.from_pretrained("Sentdex/GPyT") | |
| model = TFAutoModelForCausalLM.from_pretrained("Sentdex/GPyT", from_pt=True) | |
| tokenizer.save_pretrained(save_directory='./') | |
| model.save_pretrained(save_directory='./', saved_model=True) |