itdainb commited on
Commit
0f81abe
·
verified ·
1 Parent(s): 7e353ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -82,7 +82,7 @@ features = tokenizer(*tokenized_pairs, padding=True, truncation="longest_first",
82
 
83
  model.eval()
84
  with torch.no_grad():
85
- model_predictions = self.model(**features, return_dict=True)
86
  logits = activation_fct(model_predictions.logits)
87
 
88
  scores = [score[0] for score in logits]
 
82
 
83
  model.eval()
84
  with torch.no_grad():
85
+ model_predictions = model(**features, return_dict=True)
86
  logits = activation_fct(model_predictions.logits)
87
 
88
  scores = [score[0] for score in logits]