ypesk commited on
Commit
371be94
·
verified ·
1 Parent(s): ec69509

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +2 -1
tasks/text.py CHANGED
@@ -321,7 +321,8 @@ async def evaluate_text(request: TextEvaluationRequest):
321
  # YOUR MODEL INFERENCE CODE HERE
322
  # Update the code below to replace the random baseline by your model inference within the inference pass where the energy consumption and emissions are tracked.
323
  #--------------------------------------------------------------------------------------------
324
- predictions = []
 
325
  model.eval()
326
  for batch in tqdm(test_dataloader):
327
  batch = tuple(t.to(device) for t in batch)
 
321
  # YOUR MODEL INFERENCE CODE HERE
322
  # Update the code below to replace the random baseline by your model inference within the inference pass where the energy consumption and emissions are tracked.
323
  #--------------------------------------------------------------------------------------------
324
+
325
+ predictions = []
326
  model.eval()
327
  for batch in tqdm(test_dataloader):
328
  batch = tuple(t.to(device) for t in batch)