ViTamin-B / config.json
bbexx's picture
update
c139f9f
raw
history blame contribute delete
593 Bytes
{
"_commit_hash": null,
"architectures": [
"ViTaminCLIP"
],
"auto_map": {
"AutoConfig": "configuration_vitamin.ViTaminConfig",
"AutoModel": "model.ViTaminCLIP"
},
"embed_dim": 512,
"vision_cfg": {
"timm_model_name": "vitamin_base",
"timm_model_pretrained": false,
"timm_pool": "",
"timm_proj": "linear",
"timm_drop": 0.0,
"timm_drop_path": 0.1,
"image_size": 224
},
"text_cfg": {
"context_length": 77,
"vocab_size": 49408,
"width": 512,
"heads": 8,
"layers": 12
}
}