Downloads · 30 days
0
GPT2-GutenbergStoryGenerator
VJyzCELERY/GPT2-GutenbergStoryGenerator
GPT2-GutenbergStoryGenerator is a machine learning model from VJyzCELERY. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Refer to https://github.com/VJyzCELERY/GPT2-StoryGenerator for the model source To use the model simply do :
Access
Public
Updated Dec 10, 2025
Repo size
444 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt444 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Dec 10, 2025
- Updated
- Dec 10, 2025
- SHA
- 8db33dfe
Model card
From the Hugging Face model README
Refer to https://github.com/VJyzCELERY/GPT2-StoryGenerator for the model source To use the model simply do :
import torch
from src.model import Config,GPT
from src.inference import GPTInfer
from huggingface_hub import hf_hub_download
model_path = hf_hub_download(
repo_id="VJyzCELERY/GPT2-GutenbergStoryGenerator",
filename="GPT2-GutenbergStoryGenerator.pt"
)
checkpoint = torch.load(model_path, weights_only=False)
model = GPT(config=checkpoint['config'])
model.load_state_dict(checkpoint['model'])
model = model.to(device)
token_encoder = tiktoken.get_encoding('gpt2')
generator = GPTInfer(model, token_encoder, device)
- License
- mit
- Created
- Dec 10, 2025
- Updated
- Dec 10, 2025