razmars commited on
Commit
269387d
·
verified ·
1 Parent(s): 654d029

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +2 -2
modeling_super_linear.py CHANGED
@@ -638,9 +638,9 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
638
 
639
  if x_enc.shape[1] < 512:
640
  x_enc = self.revin_layer(x_enc, 'norm')
641
- #x_enc = self.fourier_interp_dim1(x_enc)
642
 
643
- self.backbone.inf_pred_len = 720
644
 
645
  # backbone returns (B, pred_len, C)
646
 
 
638
 
639
  if x_enc.shape[1] < 512:
640
  x_enc = self.revin_layer(x_enc, 'norm')
641
+ x_enc = self.fourier_interp_dim1(x_enc)
642
 
643
+ self.backbone.inf_pred_len = 336
644
 
645
  # backbone returns (B, pred_len, C)
646