| { |
| "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 |
| } |
| } |
| } |
|
|
