Update modeling_super_linear.py

modeling_super_linear.py

@@ -594,12 +594,12 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
         
         if x_enc.shape[1] < 512:
             x_enc = self.fourier_interp_dim1(x_enc)
-            x_enc = self.revin_layer(x_enc, 'norm')
+            #x_enc = self.revin_layer(x_enc, 'norm')
 
    
         # backbone returns (B, pred_len, C)
         preds = self.backbone(x_enc)
-        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,)