jinaai
/

jina-bert-flash-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

Markus28 commited on Feb 27

Commit

80472cb

•

1 Parent(s): 6fb6577

Fixed typo

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -294,7 +294,7 @@ class BertPreTrainingHeads(nn.Module):
         return prediction_scores, seq_relationship_score
-class BertPreTrainedModel(PretrainedModel):
     """An abstract class to handle weights initialization and
     a simple interface for dowloading and loading pretrained models.
     """

         return prediction_scores, seq_relationship_score
+class BertPreTrainedModel(PreTrainedModel):
     """An abstract class to handle weights initialization and
     a simple interface for dowloading and loading pretrained models.
     """