Pierce Maloney
commited on
Commit
•
26070b4
1
Parent(s):
320c7f3
adding second way of describing bad word (*
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -28,7 +28,7 @@ class EndpointHandler():
|
|
28 |
stopping_criteria=self.stopping_criteria,
|
29 |
max_new_tokens=50,
|
30 |
return_full_text=False,
|
31 |
-
bad_words_ids=[[3070]],
|
32 |
temperature=0.8,
|
33 |
top_k=20,
|
34 |
)
|
|
|
28 |
stopping_criteria=self.stopping_criteria,
|
29 |
max_new_tokens=50,
|
30 |
return_full_text=False,
|
31 |
+
bad_words_ids=[[3070], [313, 334]],
|
32 |
temperature=0.8,
|
33 |
top_k=20,
|
34 |
)
|