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")) | |