da03 commited on
Commit
73a674e
·
1 Parent(s): 3098025
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -43,6 +43,7 @@ def predict_product(num1, num2):
43
  outputs = model.generate(
44
  input_ids=generated_ids,
45
  max_new_tokens=1,
 
46
  past_key_values=past_key_values,
47
  return_dict_in_generate=True,
48
  use_cache=True
 
43
  outputs = model.generate(
44
  input_ids=generated_ids,
45
  max_new_tokens=1,
46
+ do_sample=False,
47
  past_key_values=past_key_values,
48
  return_dict_in_generate=True,
49
  use_cache=True