Update configuration_quiet.py

configuration_quiet.py

@@ -131,6 +131,7 @@ class QuietConfig(PretrainedConfig):
         use_complex_think_head=False,
         use_complex_talk_head=True,
         use_weighted_talk_head=True,
+        is_causal=False,  # Add the is_causal attribute
         **kwargs,
     ):
         self.vocab_size = vocab_size
@@ -162,6 +163,7 @@ class QuietConfig(PretrainedConfig):
         self.use_complex_think_head = use_complex_think_head
         self.use_complex_talk_head = use_complex_talk_head
         self.use_weighted_talk_head = use_weighted_talk_head
+        self.is_causal = is_causal  # Assign the is_causal attribute
 
         super().__init__(
             pad_token_id=pad_token_id,
@@ -169,4 +171,4 @@ class QuietConfig(PretrainedConfig):
             eos_token_id=eos_token_id,
             tie_word_embeddings=tie_word_embeddings,
             **kwargs,
-        )
+        )