numb3r3 commited on
Commit ·
6e8f0b8
1
Parent(s): 28c6280
implement compute_score api
Browse files- modeling_bert.py +1 -1
modeling_bert.py
CHANGED
|
@@ -2089,7 +2089,7 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
|
|
| 2089 |
if not hasattr(self, 'tokenizer'):
|
| 2090 |
from transformers import AutoTokenizer
|
| 2091 |
|
| 2092 |
-
self.
|
| 2093 |
|
| 2094 |
is_training = self.training
|
| 2095 |
self.eval()
|
|
|
|
| 2089 |
if not hasattr(self, 'tokenizer'):
|
| 2090 |
from transformers import AutoTokenizer
|
| 2091 |
|
| 2092 |
+
self._tokenizer = AutoTokenizer.from_pretrained(self._name_or_path)
|
| 2093 |
|
| 2094 |
is_training = self.training
|
| 2095 |
self.eval()
|