jinaai
/

jina-bert-flash-implementation

🇪🇺 Region: EU

Model card Files Files and versions

Markus28 commited on Feb 21, 2024

Commit

8c27502

·

1 Parent(s): f8b62b4

fix: try to fix import error

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -23,7 +23,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 from einops import rearrange
 from transformers import PretrainedConfig
-from configuration_bert import JinaBertConfig
 from transformers.models.bert.modeling_bert import (
     BaseModelOutputWithPoolingAndCrossAttentions,
     BertForPreTrainingOutput,

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