model_config.json

{
  "init_args": [
    {
      "hidden_size": 768,
      "bos_token_id": 57344,
      "eos_token_id": 57345,
      "pad_token_id": 0,
      "num_encoder_layers": 12,
      "num_heads": 12,
      "activation": "gelu",
      "num_hash_buckets": 16384,
      "num_hash_functions": 8,
      "type_vocab_size": 16,
      "layer_norm_eps": 1e-12,
      "upsampling_kernel_size": 4,
      "downsampling_rate": 4,
      "local_transformer_stride": 128,
      "max_position_embeddings": 16384,
      "attention_dropout": 0.1,
      "hidden_dropout": 0.1,
      "encoder_ffn_dim": 3072,
      "init_std": 0.02,
      "model_max_length": 2048,
      "init_class": "CanineModel"
    }
  ],
  "init_class": "CanineForTydiQA"
}