{ | |
"model_cfg": { | |
"embed_dim": 1024, | |
"vision_cfg": { | |
"image_size": 224, | |
"layers": 32, | |
"width": 1280, | |
"head_width": 80, | |
"patch_size": 14 | |
}, | |
"text_cfg": { | |
"hf_model_name": "facebook/nllb-200-distilled-1.3B", | |
"hf_tokenizer_name": "facebook/nllb-200-distilled-1.3B", | |
"proj": "linear", | |
"pooler_type": "cls_pooler" | |
} | |
}, | |
"preprocess_cfg": { | |
"mean": [ | |
0.48145466, | |
0.4578275, | |
0.40821073 | |
], | |
"std": [ | |
0.26862954, | |
0.26130258, | |
0.27577711 | |
] | |
} | |
} |