system HF staff commited on
Commit
c448093
1 Parent(s): 90547c2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -17
config.json CHANGED
@@ -2,18 +2,12 @@
2
  "architectures": [
3
  "RagTokenForGeneration"
4
  ],
5
- "bad_words_ids": [
6
- [
7
- 0,
8
- 0
9
- ]
10
- ],
11
  "dataset": "wiki_dpr",
12
  "dataset_split": "train",
13
- "decoder_start_token_id": 2,
 
14
  "doc_sep": " // ",
15
- "early_stopping": false,
16
- "eos_token_id": 2,
17
  "generator": {
18
  "_num_labels": 3,
19
  "activation_dropout": 0.0,
@@ -104,17 +98,14 @@
104
  "vocab_size": 50265,
105
  "xla_device": null
106
  },
107
- "index_name": "compressed",
108
  "index_path": null,
109
  "is_encoder_decoder": true,
 
110
  "max_combined_length": 300,
111
- "max_length": 50,
112
- "min_length": 1,
113
  "model_type": "rag",
114
  "n_docs": 5,
115
- "num_beams": 4,
116
- "num_return_sequences": 1,
117
- "pad_token_id": 1,
118
  "passages_path": null,
119
  "question_encoder": {
120
  "add_cross_attention": false,
@@ -179,9 +170,10 @@
179
  "vocab_size": 30522,
180
  "xla_device": null
181
  },
 
182
  "retrieval_batch_size": 8,
183
  "retrieval_vector_size": 768,
184
  "title_sep": " / ",
185
- "use_dummy_dataset": false,
186
  "vocab_size": null
187
- }
2
  "architectures": [
3
  "RagTokenForGeneration"
4
  ],
 
 
 
 
 
 
5
  "dataset": "wiki_dpr",
6
  "dataset_split": "train",
7
+ "do_deduplication": true,
8
+ "do_marginalize": false,
9
  "doc_sep": " // ",
10
+ "exclude_bos_score": false,
 
11
  "generator": {
12
  "_num_labels": 3,
13
  "activation_dropout": 0.0,
98
  "vocab_size": 50265,
99
  "xla_device": null
100
  },
101
+ "index_name": "exact",
102
  "index_path": null,
103
  "is_encoder_decoder": true,
104
+ "label_smoothing": 0.0,
105
  "max_combined_length": 300,
 
 
106
  "model_type": "rag",
107
  "n_docs": 5,
108
+ "output_retrieved": false,
 
 
109
  "passages_path": null,
110
  "question_encoder": {
111
  "add_cross_attention": false,
170
  "vocab_size": 30522,
171
  "xla_device": null
172
  },
173
+ "reduce_loss": false,
174
  "retrieval_batch_size": 8,
175
  "retrieval_vector_size": 768,
176
  "title_sep": " / ",
177
+ "use_dummy_dataset": true,
178
  "vocab_size": null
179
+ }