feat: add template json
This commit is contained in:
16
cargos/tvai-v2/models/ViT-S-16.json
Normal file
16
cargos/tvai-v2/models/ViT-S-16.json
Normal file
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"embed_dim": 384,
|
||||
"vision_cfg": {
|
||||
"image_size": 224,
|
||||
"layers": 12,
|
||||
"width": 384,
|
||||
"patch_size": 16
|
||||
},
|
||||
"text_cfg": {
|
||||
"context_length": 77,
|
||||
"vocab_size": 49408,
|
||||
"width": 384,
|
||||
"heads": 6,
|
||||
"layers": 12
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user