config.json

{
  "_name_or_path": "NRJ-350",
  "activation": "softmax",
  "alpha": 0.1,
  "architectures": [
    "BertEnergyModelForMaskedLM"
  ],
  "auto_map": {
    "AutoModel": "mlm.BertEnergyModelForMaskedLM"
  },
  "beta": 0.125,
  "bias": true,
  "block_size": 512,
  "compile": false,
  "dropout": 0.1,
  "embedding_dim": 768,
  "forward_memories": 3072,
  "layer_norm": 1e-12,
  "model_type": "bert_energy",
  "num_heads": 12,
  "num_layers": 12,
  "pad_idx": null,
  "positional": true,
  "share_layers": false,
  "tie_weights": false,
  "torch_dtype": "float32",
  "transformers_version": "4.47.0",
  "vocabulary_size": 30000
}