jinaai
/

jina-reranker-v1-tiny-en

Text Classification

Transformers.js

feature-extraction

🇪🇺 Region: EU

Model card Files Files and versions Community

numb3r3 commited on Apr 16, 2024

Commit

6e8f0b8

·

1 Parent(s): 28c6280

implement compute_score api

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -2089,7 +2089,7 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
         if not hasattr(self, 'tokenizer'):
             from transformers import AutoTokenizer
-            self.tokenizer = AutoTokenizer.from_pretrained(self._name_or_path)
         is_training = self.training
         self.eval()

         if not hasattr(self, 'tokenizer'):
             from transformers import AutoTokenizer
+            self._tokenizer = AutoTokenizer.from_pretrained(self._name_or_path)
         is_training = self.training
         self.eval()