Spaces:
Sleeping
Sleeping
Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
@@ -30,6 +30,7 @@ class SweetCommander():
|
|
30 |
input_ids["input_ids"],
|
31 |
max_new_tokens = 50,
|
32 |
temperature = 0.5,
|
|
|
33 |
top_p = 0.9,
|
34 |
top_k = 0,
|
35 |
repetition_penalty = 1.1,
|
|
|
30 |
input_ids["input_ids"],
|
31 |
max_new_tokens = 50,
|
32 |
temperature = 0.5,
|
33 |
+
do_sample = True,
|
34 |
top_p = 0.9,
|
35 |
top_k = 0,
|
36 |
repetition_penalty = 1.1,
|