zhihan1996
/

DNABERT-2-117M

Inference Endpoints

Model card Files Files and versions Community

zhihan1996 commited on Feb 22

Commit

0109f62

•

1 Parent(s): 7e37201

Update bert_layers.py

Files changed (1) hide show

bert_layers.py +1 -1

bert_layers.py CHANGED Viewed

@@ -522,7 +522,7 @@ class BertPredictionHeadTransform(nn.Module):
         return hidden_states
-class BertModel(PreTrainedModel):
     """Overall BERT model.
     Args:

         return hidden_states
+class BertModel(BertPreTrainedModel):
     """Overall BERT model.
     Args: