Update model_config.yaml
Browse files- model_config.yaml +1 -0
model_config.yaml
CHANGED
|
@@ -44,6 +44,7 @@ decoder:
|
|
| 44 |
generation:
|
| 45 |
bos_token_id: 0
|
| 46 |
decoder_start_token_id: 0
|
|
|
|
| 47 |
early_stopping: true
|
| 48 |
eos_token_id: 2
|
| 49 |
length_penalty: 2.0
|
|
|
|
| 44 |
generation:
|
| 45 |
bos_token_id: 0
|
| 46 |
decoder_start_token_id: 0
|
| 47 |
+
return_dict_in_generate: true
|
| 48 |
early_stopping: true
|
| 49 |
eos_token_id: 2
|
| 50 |
length_penalty: 2.0
|