Update modeling_super_linear.py
Browse files- modeling_super_linear.py +2 -0
modeling_super_linear.py
CHANGED
|
@@ -618,6 +618,8 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 618 |
x_enc = self.fourier_interp_dim1(x_enc)
|
| 619 |
x_enc = self.revin_layer(x_enc, 'norm')
|
| 620 |
|
|
|
|
|
|
|
| 621 |
|
| 622 |
# backbone returns (B, pred_len, C)
|
| 623 |
|
|
|
|
| 618 |
x_enc = self.fourier_interp_dim1(x_enc)
|
| 619 |
x_enc = self.revin_layer(x_enc, 'norm')
|
| 620 |
|
| 621 |
+
self.backbone.inf_pred_len = 336
|
| 622 |
+
|
| 623 |
|
| 624 |
# backbone returns (B, pred_len, C)
|
| 625 |
|