numb3r3
commited on
Commit
·
8abb27c
1
Parent(s):
cdef62a
implement compute_score api
Browse files- 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.
|
| 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()
|