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