model_config.json

{
  "init_args": [
    {
      "vocab_size": 18000,
      "hidden_size": 8,
      "num_hidden_layers": 2,
      "num_attention_heads": 2,
      "intermediate_size": 8,
      "hidden_act": "relu",
      "hidden_dropout_prob": 0.1,
      "attention_probs_dropout_prob": 0.1,
      "normalize_before": false,
      "max_position_embeddings": 513,
      "type_vocab_size": 4,
      "initializer_range": 0.02,
      "unk_token_id": 17963,
      "pad_token_id": 0,
      "bos_token_id": 1,
      "eos_token_id": 3,
      "mask_token_id": 3,
      "init_class": "UNIMOModel"
    }
  ],
  "init_class": "UNIMOLMHeadModel"
}