model_config.json

{
  "init_args": [
    {
      "attention_probs_dropout_prob": 0.1,
      "hidden_act": "gelu",
      "hidden_dropout_prob": 0.1,
      "hidden_size": 312,
      "intermediate_size": 1248,
      "initializer_range": 0.02,
      "max_position_embeddings": 2048,
      "num_attention_heads": 12,
      "num_hidden_layers": 4,
      "task_type_vocab_size": 16,
      "type_vocab_size": 4,
      "use_task_id": true,
      "vocab_size": 40000,
      "init_class": "ErnieModel"
    }
  ],
  "init_class": "UIE"
}