Export GPT in TF format using huggingface

gpt.py

from transformers import AutoTokenizer, TFOpenAIGPTModel
import tensorflow as tf

model_name = "openai-community/openai-gpt"
model_path = "/workspace/iree/workspace/gpt/gpt-tf"
cache_dir = "/workspace/iree/workspace/gpt/cache"
    
tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=cache_dir)
model = TFOpenAIGPTModel.from_pretrained(model_name, cache_dir=cache_dir)
tf.saved_model.save(model, model_path)
tokenizer.save_pretrained(model_path)