Spaces:
Running
on
Zero
Running
on
Zero
da03
commited on
Commit
·
dde88b9
1
Parent(s):
9eda610
app.py
CHANGED
|
@@ -64,7 +64,7 @@ def predict_product(num1, num2):
|
|
| 64 |
model = models[model_name]
|
| 65 |
if finished_per_model[model_name]:
|
| 66 |
continue
|
| 67 |
-
if step >=
|
| 68 |
continue
|
| 69 |
generation_kwargs = {
|
| 70 |
'input_ids': generated_ids_per_model[model_name],
|
|
|
|
| 64 |
model = models[model_name]
|
| 65 |
if finished_per_model[model_name]:
|
| 66 |
continue
|
| 67 |
+
if step >= MAX_PRODUCT_DIGITS_PER_MODEL[model_name]:
|
| 68 |
continue
|
| 69 |
generation_kwargs = {
|
| 70 |
'input_ids': generated_ids_per_model[model_name],
|