feat: add template json
This commit is contained in:
15
cargos/tvai-v2/models/nllb-clip-base.json
Normal file
15
cargos/tvai-v2/models/nllb-clip-base.json
Normal file
@@ -0,0 +1,15 @@
|
||||
{
|
||||
"embed_dim": 512,
|
||||
"vision_cfg": {
|
||||
"image_size": 224,
|
||||
"layers": 12,
|
||||
"width": 768,
|
||||
"patch_size": 32
|
||||
},
|
||||
"text_cfg": {
|
||||
"hf_model_name": "facebook/nllb-200-distilled-600M",
|
||||
"hf_tokenizer_name": "facebook/nllb-200-distilled-600M",
|
||||
"hf_proj_type": "linear",
|
||||
"hf_pooler_type": "cls_pooler"
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user