ypesk commited on
Commit
80cfa93
·
verified ·
1 Parent(s): 50d1aa6

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -171,7 +171,7 @@ async def evaluate_text(request: TextEvaluationRequest):
171
  b_texts = batch[0]
172
  logits = model(b_texts)
173
  elif MODEL == "ct":
174
- b_input_ids, b_input_mask, b_token_type_ids, b_labels = batch
175
  logits = model(b_input_ids, b_token_type_ids, b_input_mask)
176
 
177
  logits = logits.detach().cpu().numpy()
 
171
  b_texts = batch[0]
172
  logits = model(b_texts)
173
  elif MODEL == "ct":
174
+ b_input_ids, b_input_mask, b_token_type_ids = batch
175
  logits = model(b_input_ids, b_token_type_ids, b_input_mask)
176
 
177
  logits = logits.detach().cpu().numpy()