zeromodels/bart_base

1{
2  "library_name": "zeromodels",
3  "zeromodels_version": "1.3.0",
4  "model_module": "zeromodels.models.bart",
5  "model_class": "BartConditionalGenerate",
6  "variant": "bart_base",
7  "weights": "model.weights.h5",
8  "schema_version": 2,
9  "model_type": "bart",
10  "text_config": {
11    "vocab_size": 50265,
12    "hidden_dim": 768,
13    "encoder_num_layers": 6,
14    "decoder_num_layers": 6,
15    "encoder_attention_heads": 12,
16    "decoder_attention_heads": 12,
17    "encoder_ffn_dim": 3072,
18    "decoder_ffn_dim": 3072,
19    "max_position_embeddings": 1024,
20    "activation_function": "gelu",
21    "scale_embedding": false,
22    "layer_norm_eps": 1e-05,
23    "classifier_dropout": 0.0,
24    "num_labels": 3,
25    "pad_token_id": 1,
26    "bos_token_id": 0,
27    "eos_token_id": 2,
28    "decoder_start_token_id": 2
29  },
30  "generate_args": {
31    "max_new_tokens": 64
32  }
33}