File size: 248 Bytes
61ec2fe
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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"))