Update tasks/text.py
Browse files- tasks/text.py +2 -0
tasks/text.py
CHANGED
|
@@ -84,6 +84,8 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 84 |
tokenizer=tokenizer
|
| 85 |
)
|
| 86 |
|
|
|
|
|
|
|
| 87 |
preds = trainer.predict(tokenized_test)
|
| 88 |
|
| 89 |
# path_model = 'MatthiasPi/modernbert_finetunedV1'
|
|
|
|
| 84 |
tokenizer=tokenizer
|
| 85 |
)
|
| 86 |
|
| 87 |
+
trainer.args.per_device_eval_batch_size = 16
|
| 88 |
+
|
| 89 |
preds = trainer.predict(tokenized_test)
|
| 90 |
|
| 91 |
# path_model = 'MatthiasPi/modernbert_finetunedV1'
|