How to use from the
Use from the
Keras library
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"

import keras

model = keras.saving.load_model("hf://samanehs/gpt2")

** This Model Card has been manually edited **

This is a GPT2 model uploaded using the KerasNLP library. 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.

Downloads last month
5
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support