Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -640,7 +640,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 640 |
self.backbone.inf_pred_len = 96*scale_factor
|
| 641 |
|
| 642 |
preds = self.backbone(x_enc)
|
| 643 |
-
|
| 644 |
#preds = self.fourier_downsample_dim1(preds,96)
|
| 645 |
#preds = self.revin_layer(preds, 'denorm')
|
| 646 |
|
|
|
|
| 640 |
self.backbone.inf_pred_len = 96*scale_factor
|
| 641 |
|
| 642 |
preds = self.backbone(x_enc)
|
| 643 |
+
preds = self.upsample_interpolate(preds,1/scale_factor,x_enc.shape[1])
|
| 644 |
#preds = self.fourier_downsample_dim1(preds,96)
|
| 645 |
#preds = self.revin_layer(preds, 'denorm')
|
| 646 |
|