aps commited on
Commit
4285ec5
1 Parent(s): b1a60c7

Updates according to latest comments on the FLAVA PR

Browse files
Files changed (3) hide show
  1. config.json +3 -3
  2. preprocessor_config.json +0 -20
  3. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,15 +1,15 @@
1
  {
2
  "architectures": [
3
- "FLAVACodebook"
4
  ],
5
  "freeze": true,
6
  "hidden_size": 256,
7
  "initializer_range": 0.02,
8
  "input_channels": 3,
9
- "model_type": "flava_codebook",
10
  "num_blocks_per_group": 2,
11
  "num_groups": 4,
12
  "torch_dtype": "float32",
13
- "transformers_version": "4.18.0.dev0",
14
  "vocab_size": 8192
15
  }
 
1
  {
2
  "architectures": [
3
+ "FlavaImageCodebook"
4
  ],
5
  "freeze": true,
6
  "hidden_size": 256,
7
  "initializer_range": 0.02,
8
  "input_channels": 3,
9
+ "model_type": "flava_image_codebook",
10
  "num_blocks_per_group": 2,
11
  "num_groups": 4,
12
  "torch_dtype": "float32",
13
+ "transformers_version": "4.19.0.dev0",
14
  "vocab_size": 8192
15
  }
preprocessor_config.json DELETED
@@ -1,20 +0,0 @@
1
- {
2
- "crop_size": 112,
3
- "do_center_crop": true,
4
- "do_map_pixels": true,
5
- "do_normalize": true,
6
- "do_resize": true,
7
- "feature_extractor_type": "FLAVACodebookFeatureExtractor",
8
- "image_mean": [
9
- 0.0,
10
- 0.0,
11
- 0.0
12
- ],
13
- "image_std": [
14
- 1.0,
15
- 1.0,
16
- 1.0
17
- ],
18
- "resample": 1,
19
- "size": 112
20
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4c4cdf50070bcef317e0d59675bdeb639dec040524b0f9d38982ae32c1e46a4
3
- size 215167943
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a846a9283494c282f48e5f91715aa16776e4106a69bf86b900da51c186a67990
3
+ size 215168711