Update app.py
Browse files
app.py
CHANGED
@@ -61,8 +61,8 @@ def evaluate(
|
|
61 |
stream_output=False,
|
62 |
):
|
63 |
temperature=0.1
|
64 |
-
top_p=0.
|
65 |
-
top_k=
|
66 |
num_beams=1
|
67 |
max_new_tokens=380
|
68 |
|
@@ -129,6 +129,7 @@ def evaluate(
|
|
129 |
return_dict_in_generate=True,
|
130 |
output_scores=True,
|
131 |
max_new_tokens=max_new_tokens,
|
|
|
132 |
)
|
133 |
s = generation_output.sequences[0]
|
134 |
output = tokenizer.decode(s)
|
|
|
61 |
stream_output=False,
|
62 |
):
|
63 |
temperature=0.1
|
64 |
+
top_p=0.9
|
65 |
+
top_k=10
|
66 |
num_beams=1
|
67 |
max_new_tokens=380
|
68 |
|
|
|
129 |
return_dict_in_generate=True,
|
130 |
output_scores=True,
|
131 |
max_new_tokens=max_new_tokens,
|
132 |
+
seed = 42,
|
133 |
)
|
134 |
s = generation_output.sequences[0]
|
135 |
output = tokenizer.decode(s)
|