Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -102,7 +102,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 102 |
# data_collator=data_collator,
|
| 103 |
)
|
| 104 |
|
| 105 |
-
per_device_eval_batch_size=8
|
| 106 |
preds = trainer.predict(tokenized_test)
|
| 107 |
|
| 108 |
|
|
|
|
| 102 |
# data_collator=data_collator,
|
| 103 |
)
|
| 104 |
|
| 105 |
+
# per_device_eval_batch_size=8
|
| 106 |
preds = trainer.predict(tokenized_test)
|
| 107 |
|
| 108 |
|