razmars commited on
Commit
cd45014
·
verified ·
1 Parent(s): f1a7639

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. 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
- x_enc = self.upsample_interpolate(x_enc,1/scale_factor,x_enc.shape[1])
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