kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.2.4",
  "model_module": "kerasformers.models.electra",
  "model_class": "ElectraMaskedLM",
  "variant": "electra_small_generator",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "electra",
  "text_config": {
    "vocab_size": 30522,
    "embedding_size": 128,
    "embed_dim": 256,
    "num_layers": 12,
    "num_heads": 4,
    "mlp_dim": 1024,
    "max_position_embeddings": 512,
    "type_vocab_size": 2,
    "hidden_act": "gelu",
    "layer_norm_eps": 1e-12,
    "pad_token_id": 0
  }
}