feat: add template json
This commit is contained in:
14
cargos/tvai-v2/models/mt5-base-ViT-B-32.json
Normal file
14
cargos/tvai-v2/models/mt5-base-ViT-B-32.json
Normal file
@@ -0,0 +1,14 @@
|
||||
{
|
||||
"embed_dim": 512,
|
||||
"vision_cfg": {
|
||||
"image_size": 224,
|
||||
"layers": 12,
|
||||
"width": 768,
|
||||
"patch_size": 32
|
||||
},
|
||||
"text_cfg": {
|
||||
"hf_model_name": "google/mt5-base",
|
||||
"hf_tokenizer_name": "google/mt5-base",
|
||||
"hf_pooler_type": "mean_pooler"
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user