feat: add template json
This commit is contained in:
20
cargos/tvai-v2/models/ViTamin-XL-336.json
Normal file
20
cargos/tvai-v2/models/ViTamin-XL-336.json
Normal file
@@ -0,0 +1,20 @@
|
||||
{
|
||||
"embed_dim": 1152,
|
||||
"vision_cfg": {
|
||||
"timm_model_name": "vitamin_xlarge_336",
|
||||
"timm_model_pretrained": false,
|
||||
"timm_pool": "",
|
||||
"timm_proj": "linear",
|
||||
"timm_drop": 0.0,
|
||||
"timm_drop_path": 0.1,
|
||||
"image_size": 336
|
||||
},
|
||||
"text_cfg": {
|
||||
"context_length": 77,
|
||||
"vocab_size": 49408,
|
||||
"width": 1152,
|
||||
"heads": 16,
|
||||
"layers": 27
|
||||
},
|
||||
"custom_text": true
|
||||
}
|
||||
Reference in New Issue
Block a user