{
  "alphabet_size": 7,
  "attention_heads": 16,
  "attention_maps_to_save": [],
  "embed_dim": 1024,
  "embeddings_layers_to_save": [],
  "ffn_embed_dim": 2048,
  "interpolation_method": "nearest",
  "key_size": 64,
  "layer_norm_eps": 1e-05,
  "mask_token_id": 5,
  "max_positions": 20480,
  "num_cells": 10,
  "num_downsamples": 8,
  "num_hidden_layers_head": 1,
  "num_layers": 4,
  "num_scales": 10,
  "pad_token_id": 6,
  "sigma_max": 10.0,
  "sigma_min": 1.0,
  "token_embed_dim": 64,
  "use_gradient_checkpointing": false,
  "use_skip_connection": true,
  "use_spatial_information": false
}