Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
@@ -41,7 +41,7 @@
|
|
41 |
"min_length": 0,
|
42 |
"model_type": "bart",
|
43 |
"no_repeat_ngram_size": 0,
|
44 |
-
"num_beams":
|
45 |
"num_hidden_layers": 12,
|
46 |
"num_return_sequences": 1,
|
47 |
"output_attentions": false,
|
@@ -51,7 +51,7 @@
|
|
51 |
"prefix": null,
|
52 |
"pruned_heads": {},
|
53 |
"repetition_penalty": 1.0,
|
54 |
-
"scale_embedding":
|
55 |
"task_specific_params": null,
|
56 |
"temperature": 1.0,
|
57 |
"top_k": 50,
|
|
|
41 |
"min_length": 0,
|
42 |
"model_type": "bart",
|
43 |
"no_repeat_ngram_size": 0,
|
44 |
+
"num_beams": 5,
|
45 |
"num_hidden_layers": 12,
|
46 |
"num_return_sequences": 1,
|
47 |
"output_attentions": false,
|
|
|
51 |
"prefix": null,
|
52 |
"pruned_heads": {},
|
53 |
"repetition_penalty": 1.0,
|
54 |
+
"scale_embedding": true,
|
55 |
"task_specific_params": null,
|
56 |
"temperature": 1.0,
|
57 |
"top_k": 50,
|