Pierce Maloney commited on
Commit
26070b4
1 Parent(s): 320c7f3

adding second way of describing bad word (*

Browse files
Files changed (1) hide show
  1. 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
  )