numb3r3 commited on
Commit
28c6280
1 Parent(s): 1a800ed

implement compute_score api

Browse files
Files changed (1) hide show
  1. modeling_bert.py +2 -0
modeling_bert.py CHANGED
@@ -1973,6 +1973,8 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
1973
  self.num_labels = config.num_labels
1974
  self.config = config
1975
 
 
 
1976
  self.bert = JinaBertModel(config)
1977
  classifier_dropout = (
1978
  config.classifier_dropout
 
1973
  self.num_labels = config.num_labels
1974
  self.config = config
1975
 
1976
+ self._name_or_path = config._name_or_path
1977
+
1978
  self.bert = JinaBertModel(config)
1979
  classifier_dropout = (
1980
  config.classifier_dropout