Pierce Maloney commited on
Commit
37b54cb
1 Parent(s): cb274aa

temp=2! top_k 40, 1 ret seq

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