Update modeling_super_linear.py

modeling_super_linear.py

@@ -637,7 +637,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
 
         
         # backbone returns (B, pred_len, C)
-        self.backbone.inf_pred_len = 96*scale_factor
+        self.backbone.inf_pred_len = 96*2
         
         preds = self.backbone(x_enc)
         preds = self.upsample_interpolate(preds,1/scale_factor,x_enc.shape[1])