Upload model
Browse files- config.json +1 -5
config.json
CHANGED
|
@@ -152,6 +152,7 @@
|
|
| 152 |
"num_beam_groups": 1,
|
| 153 |
"num_beams": 1,
|
| 154 |
"num_channels": 3,
|
|
|
|
| 155 |
"num_return_sequences": 1,
|
| 156 |
"output_attentions": false,
|
| 157 |
"output_hidden_states": false,
|
|
@@ -172,11 +173,6 @@
|
|
| 172 |
"prefix": null,
|
| 173 |
"problem_type": null,
|
| 174 |
"pruned_heads": {},
|
| 175 |
-
"query_pool_blocks": [
|
| 176 |
-
1,
|
| 177 |
-
3,
|
| 178 |
-
10
|
| 179 |
-
],
|
| 180 |
"query_stride": [
|
| 181 |
2,
|
| 182 |
2
|
|
|
|
| 152 |
"num_beam_groups": 1,
|
| 153 |
"num_beams": 1,
|
| 154 |
"num_channels": 3,
|
| 155 |
+
"num_query_pool_stages": 3,
|
| 156 |
"num_return_sequences": 1,
|
| 157 |
"output_attentions": false,
|
| 158 |
"output_hidden_states": false,
|
|
|
|
| 173 |
"prefix": null,
|
| 174 |
"problem_type": null,
|
| 175 |
"pruned_heads": {},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 176 |
"query_stride": [
|
| 177 |
2,
|
| 178 |
2
|