added classifier dropout

configuration_bert.py

@@ -85,6 +85,7 @@ class JinaBertConfig(PretrainedConfig):
         use_flash_attn=True,
         use_qk_norm=True,
         emb_pooler=None,
+        classifier_dropout=None,
         **kwargs,
     ):
         assert 'position_embedding_type' not in kwargs
@@ -116,4 +117,5 @@ class JinaBertConfig(PretrainedConfig):
         self.num_tasks = num_tasks
         self.use_flash_attn = use_flash_attn
         self.use_qk_norm = use_qk_norm
-        self.emb_pooler = emb_pooler
+        self.emb_pooler = emb_pooler
+        self.classifier_dropout = classifier_dropout