#!/usr/bin/env python
from transformers import AutoTokenizer, TFAutoModelForCausalLM, AutoModelForCausalLM
import tensorflow as tf

tokenizer = AutoTokenizer.from_pretrained("Sentdex/GPyT")
tf_model =  TFAutoModelForCausalLM.from_pretrained("Sentdex/GPyT")
pytorch_model = AutoModelForCausalLM.from_pretrained("Sentdex/GPyT")
tokenizer.save_pretrained(save_directory='./')
tf_model.save_pretrained(save_directory='./', saved_model=True)
pytorch_model.save_pretrained(save_directory='./')