Update generation_config.json
Browse files- generation_config.json +2 -1
generation_config.json
CHANGED
|
@@ -8,5 +8,6 @@
|
|
| 8 |
"top_p": 0.8,
|
| 9 |
"cond_free_k": 2.0,
|
| 10 |
"num_return_sequences": 8,
|
| 11 |
-
"max_new_tokens": 256
|
|
|
|
| 12 |
}
|
|
|
|
| 8 |
"top_p": 0.8,
|
| 9 |
"cond_free_k": 2.0,
|
| 10 |
"num_return_sequences": 8,
|
| 11 |
+
"max_new_tokens": 256,
|
| 12 |
+
"return_dict_in_generate": true
|
| 13 |
}
|