{
  "library_name": "neurovlm",
  "model_type": "neurovlm_projection_heads",
  "projection_heads": {
    "proj_head_image_infonce": {
      "class": "neurovlm.models.ProjHead",
      "file": "proj_head_image_infonce.safetensors",
      "hidden_dim": 384,
      "latent_in_dim": 384,
      "latent_out_dim": 384
    },
    "proj_head_text_infonce": {
      "class": "neurovlm.models.ProjHead",
      "file": "proj_head_text_infonce.safetensors",
      "hidden_dim": 512,
      "latent_in_dim": 768,
      "latent_out_dim": 384
    },
    "proj_head_text_mse": {
      "class": "neurovlm.models.ProjHead",
      "file": "proj_head_text_mse.safetensors",
      "hidden_dim": 512,
      "latent_in_dim": 768,
      "latent_out_dim": 384
    }
  }
}