Spaces:
Running
on
Zero
Running
on
Zero
da03
commited on
Commit
·
73a674e
1
Parent(s):
3098025
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
|