gemma-essays / README.md
sparsh35's picture
Upload folder using huggingface_hub
754b568 verified
metadata
library_name: keras-nlp
pipeline_tag: text-generation

This is a Gemma 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: gemma_backbone
  • trainable: True
  • vocabulary_size: 256000
  • num_layers: 28
  • num_query_heads: 16
  • num_key_value_heads: 16
  • hidden_dim: 3072
  • intermediate_dim: 49152
  • head_dim: 256
  • layer_norm_epsilon: 1e-06
  • dropout: 0
  • query_head_dim_normalize: True
  • use_post_ffw_norm: False
  • use_post_attention_norm: False
  • final_logit_soft_cap: None
  • attention_logit_soft_cap: None
  • sliding_window_size: 4096
  • use_sliding_window_attention: False

This model card has been generated automatically and should be completed by the model author. See Model Cards documentation for more information.