zhihan1996
/

DNABERT-2-117M

Model card Files Files and versions

zhihan1996 commited on Feb 23, 2024

Commit

dd10f74

·

verified ·

1 Parent(s): 1ce72a5

Update bert_layers.py

Files changed (1) hide show

bert_layers.py +1 -1

bert_layers.py CHANGED Viewed

@@ -18,7 +18,7 @@ from torch.nn.modules.utils import consume_prefix_in_state_dict_if_present
 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import (MaskedLMOutput,
                                            SequenceClassifierOutput)
-from transformers.modeling_bert import BertPreTrainedModel
 from transformers.modeling_utils import PreTrainedModel
 from .bert_padding import (index_first_axis,

 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import (MaskedLMOutput,
                                            SequenceClassifierOutput)
+from transformers.models.bert.modeling_bert import BertPreTrainedModel
 from transformers.modeling_utils import PreTrainedModel
 from .bert_padding import (index_first_axis,