Downloads · 30 days
18
2% of all-time downloads
text-generation-miniature-gpt
keras-io/text-generation-miniature-gpt
text-generation-miniature-gpt is a text generation model from keras-io. Use it when you need the model to write or continue text. It is set up for tf-keras. The card lists the license as gpl.
This repo contains the model and the notebook to this Keras example on Text generation with a miniature GPT.
All-time downloads
1.1K
Public
Repo size
264 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.data-00000-of-00001131 MB · 100%
At a glance
- Task
- Text Generation
- Library
- tf-keras
- License
- gpl
- Model type
- gpt2
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Feb 21, 2022
- SHA
- 6463786d
Try a prompt
Spaces
Model card
From the Hugging Face model README
Keras Implementation of Text generation with a miniature GPT
This repo contains the model and the notebook to this Keras example on Text generation with a miniature GPT.
Full credits to: fchollet
Background Information
This example demonstrates how to implement text generation with a miniature GPT model. The model consists of a single Transformer block with causal masking in its attention layer.
Datasets
IMDB sentiment classification dataset for training. The model generates new movie reviews for a given prompt.
- Task
- Text Generation
- Library
- tf-keras
- Type
- gpt2
- License
- gpl
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Feb 21, 2022