Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -90,7 +90,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 90 |
data_collator=data_collator
|
| 91 |
)
|
| 92 |
|
| 93 |
-
trainer.args.per_device_eval_batch_size =
|
| 94 |
|
| 95 |
preds = trainer.predict(tokenized_test)
|
| 96 |
|
|
|
|
| 90 |
data_collator=data_collator
|
| 91 |
)
|
| 92 |
|
| 93 |
+
trainer.args.per_device_eval_batch_size = 16
|
| 94 |
|
| 95 |
preds = trainer.predict(tokenized_test)
|
| 96 |
|