Update modeling_super_linear.py

modeling_super_linear.py

@@ -474,7 +474,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
                 labels: Optional[torch.Tensor] = None,        
                 **kwargs,) -> CausalLMOutputWithCrossAttentions:
 
-
+        
         if inputs_embeds is None:
             raise ValueError("Pass the time‑series as `inputs_embeds`")
         
@@ -482,7 +482,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
         x_enc = inputs_embeds
     
         # backbone returns (B, pred_len, C)
-        preds = self.backbone(x_enc)
+        preds = self.backbone(x_enc, pred_len=kwargs.get("pred_len", default_value))
         return CausalLMOutputWithCrossAttentions(loss=None,logits=preds,past_key_values=None,hidden_states=None,attentions=None,)