IMvision12 commited on
Commit
60f8da1
·
verified ·
1 Parent(s): ef134e8

Upgrade kf_config.json to v2 (nested config + dtype)

Browse files
Files changed (1) hide show
  1. kf_config.json +19 -13
kf_config.json CHANGED
@@ -1,23 +1,29 @@
1
  {
2
  "library_name": "kerasformers",
3
- "kerasformers_version": "1.1.3",
4
  "model_module": "kerasformers.models.owlv2",
5
  "model_class": "Owlv2Detect",
6
  "variant": "owlv2-large-patch14-ensemble",
7
  "weights": "model.weights.h5",
 
 
8
  "model_type": "owlv2",
9
- "vision_image_size": 1008,
10
- "vision_patch_size": 14,
11
- "vision_hidden_dim": 1024,
12
- "vision_intermediate_size": 4096,
13
- "vision_num_layers": 24,
14
- "vision_num_heads": 16,
15
- "text_hidden_dim": 768,
16
- "text_intermediate_size": 3072,
17
- "text_num_heads": 16,
 
 
 
 
 
 
 
18
  "projection_dim": 768,
19
- "text_num_layers": 12,
20
- "text_max_position_embeddings": 16,
21
- "text_vocab_size": 49408,
22
  "image_size": null
23
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.owlv2",
5
  "model_class": "Owlv2Detect",
6
  "variant": "owlv2-large-patch14-ensemble",
7
  "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "dtype": "float32",
10
  "model_type": "owlv2",
11
+ "text_config": {
12
+ "hidden_dim": 768,
13
+ "intermediate_size": 3072,
14
+ "num_heads": 16,
15
+ "num_layers": 12,
16
+ "max_position_embeddings": 16,
17
+ "vocab_size": 49408
18
+ },
19
+ "vision_config": {
20
+ "image_size": 1008,
21
+ "patch_size": 14,
22
+ "hidden_dim": 1024,
23
+ "intermediate_size": 4096,
24
+ "num_layers": 24,
25
+ "num_heads": 16
26
+ },
27
  "projection_dim": 768,
 
 
 
28
  "image_size": null
29
  }