File size: 554 Bytes
ebc6d95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import TFGPT2LMHeadModel, GPT2Config, GPT2LMHeadModel
# Load your trained PyTorch model
pytorch_model_path = "trained_path"
config = GPT2Config.from_pretrained(pytorch_model_path)
pytorch_model = GPT2LMHeadModel.from_pretrained(pytorch_model_path, config=config,from_tf=True)
# Convert to TensorFlow model
tf_model = TFGPT2LMHeadModel.from_pretrained(pytorch_model_path, from_pt=True, config=config)
# Save the TensorFlow model
tf_model.save_pretrained(pytorch_model_path) # This will generate the tf_model.h5 file in the directory
|