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],
|