Update tasks/text.py

tasks/text.py

@@ -108,8 +108,6 @@ async def evaluate_text(request: TextEvaluationRequest):
     predictions = []
     for step, batch in enumerate(test_dataloader):
 
-        # Add batch to GPU
-        batch = tuple(t.to(device) for t in batch)
 
         b_input_ids, b_input_mask, b_token_type_ids = batch
         with torch.no_grad():