TenzinGayche commited on
Commit
ce6fe13
·
verified ·
1 Parent(s): f3bf7cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -65,6 +65,9 @@ def generate(
65
  {"input_ids": input_ids},
66
  streamer=streamer,
67
  max_new_tokens=max_new_tokens,
 
 
 
68
  )
69
 
70
  # Run generation in a background thread
 
65
  {"input_ids": input_ids},
66
  streamer=streamer,
67
  max_new_tokens=max_new_tokens,
68
+ top_p=top_p,
69
+ do_sample=do_sample,
70
+ temperature=temperature
71
  )
72
 
73
  # Run generation in a background thread