zm_config.json

{
  "library_name": "zeromodels",
  "zeromodels_version": "1.3.0",
  "model_module": "zeromodels.models.bart",
  "model_class": "BartConditionalGenerate",
  "variant": "bart_base",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "model_type": "bart",
  "text_config": {
    "vocab_size": 50265,
    "hidden_dim": 768,
    "encoder_num_layers": 6,
    "decoder_num_layers": 6,
    "encoder_attention_heads": 12,
    "decoder_attention_heads": 12,
    "encoder_ffn_dim": 3072,
    "decoder_ffn_dim": 3072,
    "max_position_embeddings": 1024,
    "activation_function": "gelu",
    "scale_embedding": false,
    "layer_norm_eps": 1e-05,
    "classifier_dropout": 0.0,
    "num_labels": 3,
    "pad_token_id": 1,
    "bos_token_id": 0,
    "eos_token_id": 2,
    "decoder_start_token_id": 2
  },
  "generate_args": {
    "max_new_tokens": 64
  }
}