LopezLLM / tf_model.h5
hollywoodfrancis's picture
Create tf_model.h5
61ec2fe verified
raw
history blame
248 Bytes
from transformers import TFGPT2LMHeadModel
# Load model
tf_model = TFGPT2LMHeadModel.from_pretrained("gpt2")
# Save TensorFlow model
tf_model.save_pretrained("LopezLLM", saved_model=False)
import os
print("Saved files:", os.listdir("LopezLLM"))