Spaces:
Running
on
Zero
Running
on
Zero
da03
commited on
Commit
•
e01b57c
1
Parent(s):
df028fe
app.py
CHANGED
|
@@ -76,7 +76,7 @@ def predict_product(num1, num2):
|
|
| 76 |
outputs = model.generate(**generation_kwargs)
|
| 77 |
generated_ids = outputs.sequences
|
| 78 |
next_token_id = generated_ids[0, -1]
|
| 79 |
-
print (next_token_id)
|
| 80 |
|
| 81 |
if next_token_id.item() == tokenizer.eos_token_id:
|
| 82 |
finished_per_model[model_name] = True
|
|
|
|
| 76 |
outputs = model.generate(**generation_kwargs)
|
| 77 |
generated_ids = outputs.sequences
|
| 78 |
next_token_id = generated_ids[0, -1]
|
| 79 |
+
#print (next_token_id)
|
| 80 |
|
| 81 |
if next_token_id.item() == tokenizer.eos_token_id:
|
| 82 |
finished_per_model[model_name] = True
|