Pierce Maloney commited on
Commit
320c7f3
1 Parent(s): 4f7f1d5

increase temp, add top_k=20

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -29,7 +29,8 @@ class EndpointHandler():
29
  max_new_tokens=50,
30
  return_full_text=False,
31
  bad_words_ids=[[3070]],
32
- temperature=0.7
 
33
  )
34
  return prediction
35
 
 
29
  max_new_tokens=50,
30
  return_full_text=False,
31
  bad_words_ids=[[3070]],
32
+ temperature=0.8,
33
+ top_k=20,
34
  )
35
  return prediction
36