Update configuration_neural_sbs.py

configuration_neural_sbs.py

@@ -11,10 +11,16 @@ class NeuralSBSConfig(PretrainedConfig):
 
     def __init__(
         self,
-        backbone_name="facebook/hubert-base-ls960",
-        embedding_dim=768,
+        backbone_name: str = "facebook/hubert-base-ls960",
+        bert_model_name: str = "google-bert/bert-base-uncased",
+        embedding_dim: int = 768,
+        attention_dim: int = 512,
+        max_text_seq_length: int = 512,
         **kwargs,
     ):
         super().__init__(**kwargs)
         self.backbone_name = backbone_name
-        self.embedding_dim = embedding_dim
+        self.bert_model_name = bert_model_name
+        self.embedding_dim = embedding_dim
+        self.attention_dim = attention_dim
+        self.max_text_seq_length = max_text_seq_length