Commit
•
e02d90c
1
Parent(s):
04d5c0d
Final config
Browse files- config.json +5 -0
config.json
CHANGED
|
@@ -43,9 +43,14 @@
|
|
| 43 |
"LABEL_1": 1,
|
| 44 |
"LABEL_2": 2
|
| 45 |
},
|
|
|
|
| 46 |
"max_decoder_position_embeddings": 1024,
|
| 47 |
"max_encoder_position_embeddings": 16384,
|
|
|
|
|
|
|
| 48 |
"model_type": "led",
|
|
|
|
|
|
|
| 49 |
"num_hidden_layers": 6,
|
| 50 |
"pad_token_id": 1,
|
| 51 |
"torch_dtype": "float32",
|
|
|
|
| 43 |
"LABEL_1": 1,
|
| 44 |
"LABEL_2": 2
|
| 45 |
},
|
| 46 |
+
"length_penalty": 2,
|
| 47 |
"max_decoder_position_embeddings": 1024,
|
| 48 |
"max_encoder_position_embeddings": 16384,
|
| 49 |
+
"min_new_tokens": 100,
|
| 50 |
+
"max_new_tokens": 1024,
|
| 51 |
"model_type": "led",
|
| 52 |
+
"no_repeat_ngram_size": 3,
|
| 53 |
+
"num_beams": 5,
|
| 54 |
"num_hidden_layers": 6,
|
| 55 |
"pad_token_id": 1,
|
| 56 |
"torch_dtype": "float32",
|