nobug12345 commited on
Commit
552fd82
·
verified ·
1 Parent(s): f1bffc7

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -11
config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "architectures": [
3
  "VisionEncoderDecoderModel"
4
  ],
@@ -20,7 +21,7 @@
20
  "decoder_attention_heads": 16,
21
  "decoder_ffn_dim": 4096,
22
  "decoder_layerdrop": 0.0,
23
- "decoder_layers": 8,
24
  "decoder_start_token_id": null,
25
  "diversity_penalty": 0.0,
26
  "do_sample": false,
@@ -49,7 +50,7 @@
49
  },
50
  "length_penalty": 1.0,
51
  "max_length": 20,
52
- "max_position_embeddings": 1536,
53
  "min_length": 0,
54
  "model_type": "mbart",
55
  "no_repeat_ngram_size": 0,
@@ -123,8 +124,8 @@
123
  "1": "LABEL_1"
124
  },
125
  "image_size": [
126
- 420,
127
- 420
128
  ],
129
  "initializer_range": 0.02,
130
  "is_decoder": false,
@@ -156,7 +157,6 @@
156
  "output_scores": false,
157
  "pad_token_id": null,
158
  "patch_size": 4,
159
- "path_norm": true,
160
  "prefix": null,
161
  "problem_type": null,
162
  "pruned_heads": {},
@@ -178,16 +178,14 @@
178
  "torch_dtype": null,
179
  "torchscript": false,
180
  "typical_p": 1.0,
181
- "use_2d_embeddings": false,
182
  "use_absolute_embeddings": false,
183
  "use_bfloat16": false,
184
- "window_size": 5
185
  },
186
  "is_encoder_decoder": true,
187
  "model_type": "vision-encoder-decoder",
188
  "pad_token_id": 1,
189
  "tie_word_embeddings": false,
190
- "torch_dtype": "float16",
191
- "transformers_version": "4.36.0"
192
- }
193
-
 
1
  {
2
+ "_name_or_path": ".",
3
  "architectures": [
4
  "VisionEncoderDecoderModel"
5
  ],
 
21
  "decoder_attention_heads": 16,
22
  "decoder_ffn_dim": 4096,
23
  "decoder_layerdrop": 0.0,
24
+ "decoder_layers": 4,
25
  "decoder_start_token_id": null,
26
  "diversity_penalty": 0.0,
27
  "do_sample": false,
 
50
  },
51
  "length_penalty": 1.0,
52
  "max_length": 20,
53
+ "max_position_embeddings": 3584,
54
  "min_length": 0,
55
  "model_type": "mbart",
56
  "no_repeat_ngram_size": 0,
 
124
  "1": "LABEL_1"
125
  },
126
  "image_size": [
127
+ 896,
128
+ 672
129
  ],
130
  "initializer_range": 0.02,
131
  "is_decoder": false,
 
157
  "output_scores": false,
158
  "pad_token_id": null,
159
  "patch_size": 4,
 
160
  "prefix": null,
161
  "problem_type": null,
162
  "pruned_heads": {},
 
178
  "torch_dtype": null,
179
  "torchscript": false,
180
  "typical_p": 1.0,
 
181
  "use_absolute_embeddings": false,
182
  "use_bfloat16": false,
183
+ "window_size": 7
184
  },
185
  "is_encoder_decoder": true,
186
  "model_type": "vision-encoder-decoder",
187
  "pad_token_id": 1,
188
  "tie_word_embeddings": false,
189
+ "torch_dtype": "float32",
190
+ "transformers_version": "4.35.0"
191
+ }