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()
|