Update modeling_super_linear.py

modeling_super_linear.py

@@ -639,7 +639,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
         self.backbone.inf_pred_len = 96*2
         
         preds = self.backbone(x_enc)
-        preds = self.upsample_interpolate(preds,1/scale_factor,x_enc.shape[1])
+        preds = self.upsample_interpolate(preds,1/scale_factor,96)
         print(preds.shape)
         #preds = self.fourier_downsample_dim1(preds,96)
         #preds = self.revin_layer(preds, 'denorm')