config.json

{
  "auto_map": {
    "AutoConfig": "configuration_ax_ve.AXVEConfig"
  },
  "vision_config": {
    "drop_path_rate": 0.0,
    "hidden_size": 1152,
    "image_size": 384,
    "intermediate_size": 4304,
    "num_position_embeddings": 576,
    "model_type": "ax_ve",
    "num_attention_heads": 16,
    "num_hidden_layers": 27,
    "torch_dtype": "bfloat16",
    "spatial_merge_size": 2
  }
}