feat: add template json
This commit is contained in:
18
cargos/tvai-v2/models/nllb-clip-large-siglip.json
Normal file
18
cargos/tvai-v2/models/nllb-clip-large-siglip.json
Normal file
@@ -0,0 +1,18 @@
|
||||
{
|
||||
"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"
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user