jinaai
/

jina-bert-v2-qk-post-norm

🇪🇺 Region: EU

Model card Files Files and versions Community

bwang0911 commited on Feb 6

Commit

099544e

•

1 Parent(s): bc83a13

Update configuration_bert.py

Files changed (1) hide show

configuration_bert.py +1 -1

configuration_bert.py CHANGED Viewed

@@ -128,7 +128,7 @@ class JinaBertConfig(PretrainedConfig):
         classifier_dropout=None,
         feed_forward_type="original",
         emb_pooler=None,
-        attn_implementation='torch',
         **kwargs,
     ):
         super().__init__(pad_token_id=pad_token_id, **kwargs)

         classifier_dropout=None,
         feed_forward_type="original",
         emb_pooler=None,
+        attn_implementation=None,
         **kwargs,
     ):
         super().__init__(pad_token_id=pad_token_id, **kwargs)