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.
- Downloads last month
- 17
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the HF Inference API does not support tf-keras models with pipeline type text-generation