{ | |
"model_cfg": { | |
"embed_dim": 1152, | |
"custom_text": true, | |
"init_logit_bias": -10, | |
"vision_cfg": { | |
"image_size": 384, | |
"timm_model_name": "vit_so400m_patch14_siglip_384", | |
"timm_model_pretrained": false, | |
"timm_pool": "map", | |
"timm_proj": "none" | |
}, | |
"text_cfg": { | |
"hf_model_name": "facebook/nllb-200-distilled-1.3B", | |
"hf_tokenizer_name": "facebook/nllb-200-distilled-1.3B", | |
"hf_proj_type": "linear", | |
"hf_pooler_type": "cls_pooler" | |
} | |
}, | |
"preprocess_cfg": { | |
"mean": [ | |
0.5, | |
0.5, | |
0.5 | |
], | |
"std": [ | |
0.5, | |
0.5, | |
0.5 | |
], | |
"interpolation": "bicubic", | |
"resize_mode": "squash" | |
} | |
} |