laurenssam commited on
Commit
36de391
·
1 Parent(s): 9f7caae
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -170,10 +170,9 @@ def get_response(params):
170
  # RuntimeError: The size of tensor a (2) must match the size of tensor b (3) at non-singleton dimension 0
171
  generate_kwargs = dict(
172
  inputs=input_ids,
173
- do_sample=do_sample,
174
- temperature=temperature,
175
- top_p=top_p,
176
- max_new_tokens=max_new_tokens,
177
  pad_token_id=tokenizer.pad_token_id,
178
  streamer=streamer,
179
  # stopping_criteria=[stopping_criteria],
 
170
  # RuntimeError: The size of tensor a (2) must match the size of tensor b (3) at non-singleton dimension 0
171
  generate_kwargs = dict(
172
  inputs=input_ids,
173
+ do_sample=False,
174
+ top_p=1.0,
175
+ max_new_tokens=512,
 
176
  pad_token_id=tokenizer.pad_token_id,
177
  streamer=streamer,
178
  # stopping_criteria=[stopping_criteria],