SivilTaram
commited on
Commit
•
95b0ad4
1
Parent(s):
5c84557
Fix the bug in generation config
Browse filesIf we follow the default generation configuration, it does not utilize the key-value cache during inference. This can cause the model to be too slow to generate text efficiently.
- generation_config.json +1 -1
generation_config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
"_from_model_config": true,
|
3 |
"transformers_version": "4.37.2",
|
4 |
-
"use_cache":
|
5 |
}
|
|
|
1 |
{
|
2 |
"_from_model_config": true,
|
3 |
"transformers_version": "4.37.2",
|
4 |
+
"use_cache": true
|
5 |
}
|