lirannoc commited on
Commit
628d649
·
verified ·
1 Parent(s): 51b6e9f

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -573,7 +573,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
573
  raise ValueError("Pass the time‑series as `inputs_embeds`")
574
 
575
  # backbone expects (B, C, L)
576
- preds = self.backbone(inputs_embeds, inf_pred_len = prediction_len)
577
  return CausalLMOutputWithCrossAttentions(loss=None,logits=preds,past_key_values=None,hidden_states=None,attentions=None,)
578
 
579
 
 
573
  raise ValueError("Pass the time‑series as `inputs_embeds`")
574
 
575
  # backbone expects (B, C, L)
576
+ preds = self.backbone(inputs_embeds)
577
  return CausalLMOutputWithCrossAttentions(loss=None,logits=preds,past_key_values=None,hidden_states=None,attentions=None,)
578
 
579