Spaces:
Running
on
T4
Running
on
T4
MatthiasPicard
commited on
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -84,7 +84,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
84 |
tokenizer=tokenizer
|
85 |
)
|
86 |
|
87 |
-
trainer.args.per_device_eval_batch_size =
|
88 |
|
89 |
preds = trainer.predict(tokenized_test)
|
90 |
|
|
|
84 |
tokenizer=tokenizer
|
85 |
)
|
86 |
|
87 |
+
trainer.args.per_device_eval_batch_size = 4
|
88 |
|
89 |
preds = trainer.predict(tokenized_test)
|
90 |
|