Update configuration_auristream.py

configuration_auristream.py

@@ -12,6 +12,7 @@ class AuriStreamConfig(PretrainedConfig):
         dropout=0.0,
         bias=False,
         use_rope=True,
+        skip_connections=True,
         n_pred_steps=20,
         **kwargs
     ):
@@ -22,5 +23,6 @@ class AuriStreamConfig(PretrainedConfig):
         self.dropout = dropout
         self.bias = bias
         self.use_rope = use_rope
+        self.skip_connections = skip_connections
         self.n_pred_steps = n_pred_steps
         super().__init__(**kwargs)