Markus28 commited on
Commit
75a4e4d
1 Parent(s): c2d8dc3

feat: added dense_seq_output to config

Browse files
Files changed (1) hide show
  1. configuration_bert.py +2 -0
configuration_bert.py CHANGED
@@ -74,6 +74,7 @@ class JinaBertConfig(PretrainedConfig):
74
  layer_norm_eps=1e-12,
75
  pad_token_id=0,
76
  window_size=(-1, -1),
 
77
  **kwargs,
78
  ):
79
  super().__init__(pad_token_id=pad_token_id, **kwargs)
@@ -90,4 +91,5 @@ class JinaBertConfig(PretrainedConfig):
90
  self.initializer_range = initializer_range
91
  self.layer_norm_eps = layer_norm_eps
92
  self.window_size = window_size
 
93
 
 
74
  layer_norm_eps=1e-12,
75
  pad_token_id=0,
76
  window_size=(-1, -1),
77
+ dense_seq_output=False,
78
  **kwargs,
79
  ):
80
  super().__init__(pad_token_id=pad_token_id, **kwargs)
 
91
  self.initializer_range = initializer_range
92
  self.layer_norm_eps = layer_norm_eps
93
  self.window_size = window_size
94
+ self.dense_seq_output = dense_seq_output
95