gpt2_imdb / README.md
swritchie's picture
Upload folder using huggingface_hub
f465717 verified
|
raw
history blame
700 Bytes
---
library_name: keras-nlp
pipeline_tag: text-generation
---
This is a [`GPT2` model](https://keras.io/api/keras_nlp/models/gpt2) 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](https://huggingface.co/docs/hub/model-cards) for more information.