jinaai
/

jina-bert-flash-implementation

🇪🇺 Region: EU

Model card Files Files and versions

Markus28 commited on Feb 27, 2024

Commit

6fb6577

·

1 Parent(s): e209593

Fixed typo

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -22,7 +22,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from einops import rearrange
-from transformers.modeling_utils import PretrainedModel
 from .configuration_bert import JinaBertConfig
 from transformers.models.bert.modeling_bert import (
     BaseModelOutputWithPoolingAndCrossAttentions,

 import torch.nn as nn
 import torch.nn.functional as F
 from einops import rearrange
+from transformers.modeling_utils import PreTrainedModel
 from .configuration_bert import JinaBertConfig
 from transformers.models.bert.modeling_bert import (
     BaseModelOutputWithPoolingAndCrossAttentions,