Spaces:
Runtime error
Runtime error
change generation config a bit
Browse files
app.py
CHANGED
|
@@ -57,9 +57,9 @@ class Pipeline:
|
|
| 57 |
self.generation_config = {
|
| 58 |
"max_new_tokens": 1024,
|
| 59 |
"top_k": 40,
|
| 60 |
-
"top_p": 0.
|
| 61 |
-
"temperature": 0.
|
| 62 |
-
"repetition_penalty": 1.
|
| 63 |
"last_n_tokens": 64,
|
| 64 |
"seed": -1,
|
| 65 |
"batch_size": 8,
|
|
|
|
| 57 |
self.generation_config = {
|
| 58 |
"max_new_tokens": 1024,
|
| 59 |
"top_k": 40,
|
| 60 |
+
"top_p": 0.90,
|
| 61 |
+
"temperature": 0.72,
|
| 62 |
+
"repetition_penalty": 1.22,
|
| 63 |
"last_n_tokens": 64,
|
| 64 |
"seed": -1,
|
| 65 |
"batch_size": 8,
|