Update modeling_super_linear.py

modeling_super_linear.py

@@ -633,8 +633,8 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
                 
 
         
-        #if x_enc.shape[1] < 512:
-            #x_enc = self.revin_layer(x_enc, 'norm')
+        if x_enc.shape[1] < 512:
+            x_enc = self.revin_layer(x_enc, 'norm')
             #x_enc = self.fourier_interp_dim1(x_enc)
 
         #self.backbone.inf_pred_len = 336
@@ -643,7 +643,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
         
         preds = self.backbone(x_enc)
         #preds = self.fourier_downsample_dim1(preds,96)
-        #preds = self.revin_layer(preds, 'denorm')
+        preds = self.revin_layer(preds, 'denorm')
                 
         return CausalLMOutputWithCrossAttentions(loss=None,logits=preds,past_key_values=None,hidden_states=None,attentions=None,)