Downloads · 30 days
15
31% of all-time downloads
gpt2_imdb
swritchie/gpt2_imdb
gpt2_imdb is a text generation model from swritchie. Use it when you need the model to write or continue text. It is set up for keras-nlp.
This is a GPT2 model uploaded using the KerasNLP library and can be used with JAX, TensorFlow, and PyTorch backends. This model is related to a CausalLM task.
All-time downloads
49
Public
Repo size
498 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.h5498 MB · 100%
At a glance
- Task
- Text Generation
- Library
- keras-nlp
- Access
- Public
- Created
- Aug 21, 2024
- Updated
- Aug 21, 2024
- SHA
- f4657172
Try a prompt
Model card
From the Hugging Face model README
This is a GPT2 model uploaded using the KerasNLP library and can be used with JAX, TensorFlow, and PyTorch backends.
This model is related to a CausalLM task.
Model config:
- name: gpt2_backbone
- trainable: True
- vocabulary_size: 50257
- num_layers: 12
- num_heads: 12
- hidden_dim: 768
- intermediate_dim: 3072
- dropout: 0.1
- max_sequence_length: 1024
This model card has been generated automatically and should be completed by the model author. See Model Cards documentation for more information.
- Task
- Text Generation
- Library
- keras-nlp
- Created
- Aug 21, 2024
- Updated
- Aug 21, 2024