Transformers
PyTorch
rag
Inference Endpoints
system HF staff commited on
Commit
d7a0861
1 Parent(s): 39dee4d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -14
config.json CHANGED
@@ -2,16 +2,12 @@
2
  "architectures": [
3
  "RagSequenceForGeneration"
4
  ],
5
- "bad_words_ids": [
6
- [
7
- 0,
8
- 0
9
- ]
10
- ],
11
  "dataset": "wiki_dpr",
12
  "dataset_split": "train",
 
 
13
  "doc_sep": " // ",
14
- "early_stopping": false,
15
  "generator": {
16
  "_num_labels": 3,
17
  "activation_dropout": 0.0,
@@ -102,16 +98,14 @@
102
  "vocab_size": 50265,
103
  "xla_device": null
104
  },
105
- "index_name": "compressed",
106
  "index_path": null,
107
  "is_encoder_decoder": true,
 
108
  "max_combined_length": 300,
109
- "max_length": 50,
110
- "min_length": 1,
111
  "model_type": "rag",
112
  "n_docs": 5,
113
- "num_beams": 4,
114
- "num_return_sequences": 1,
115
  "passages_path": null,
116
  "question_encoder": {
117
  "add_cross_attention": false,
@@ -176,9 +170,10 @@
176
  "vocab_size": 30522,
177
  "xla_device": null
178
  },
 
179
  "retrieval_batch_size": 8,
180
  "retrieval_vector_size": 768,
181
  "title_sep": " / ",
182
- "use_dummy_dataset": false,
183
  "vocab_size": null
184
- }
 
2
  "architectures": [
3
  "RagSequenceForGeneration"
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
+ }