Commit
·
152e2e7
1
Parent(s):
9015e74
Update config.json
Browse files- config.json +7 -0
config.json
CHANGED
|
@@ -52,6 +52,13 @@
|
|
| 52 |
"max_length": 300,
|
| 53 |
"num_beams": 4,
|
| 54 |
"prefix": "translate English to Romanian: "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
}
|
| 56 |
},
|
| 57 |
"torch_dtype": "float32",
|
|
|
|
| 52 |
"max_length": 300,
|
| 53 |
"num_beams": 4,
|
| 54 |
"prefix": "translate English to Romanian: "
|
| 55 |
+
},
|
| 56 |
+
"text-generation": {
|
| 57 |
+
"early_stopping": true,
|
| 58 |
+
"do_sample": true,
|
| 59 |
+
"max_length": 128,
|
| 60 |
+
"num_beams": 5,
|
| 61 |
+
"num_return_sequences": 2
|
| 62 |
}
|
| 63 |
},
|
| 64 |
"torch_dtype": "float32",
|