Pierce Maloney
commited on
Commit
•
37b54cb
1
Parent(s):
cb274aa
temp=2! top_k 40, 1 ret seq
Browse files- handler.py +1 -2
handler.py
CHANGED
@@ -29,9 +29,8 @@ class EndpointHandler():
|
|
29 |
max_new_tokens=50,
|
30 |
return_full_text=False,
|
31 |
bad_words_ids=[[3070], [313, 334]],
|
32 |
-
temperature=
|
33 |
top_k=40,
|
34 |
-
num_return_sequences=2
|
35 |
)
|
36 |
return prediction
|
37 |
|
|
|
29 |
max_new_tokens=50,
|
30 |
return_full_text=False,
|
31 |
bad_words_ids=[[3070], [313, 334]],
|
32 |
+
temperature=2,
|
33 |
top_k=40,
|
|
|
34 |
)
|
35 |
return prediction
|
36 |
|