Update configuration_auristream.py

configuration_auristream.py

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