numb3r3 commited on
Commit
8abb27c
1 Parent(s): cdef62a

implement compute_score api

Browse files
Files changed (1) hide show
  1. modeling_bert.py +1 -1
modeling_bert.py CHANGED
@@ -2059,7 +2059,7 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
2059
  if not hasattr(self, 'tokenizer'):
2060
  from transformers import AutoTokenizer
2061
 
2062
- self.tokenizer = AutoTokenizer.from_pretrained(self._name_or_path)
2063
 
2064
  is_training = self.training
2065
  self.eval()
 
2059
  if not hasattr(self, 'tokenizer'):
2060
  from transformers import AutoTokenizer
2061
 
2062
+ self._tokenizer = AutoTokenizer.from_pretrained(self._name_or_path)
2063
 
2064
  is_training = self.training
2065
  self.eval()