Update modeling_super_linear.py

modeling_super_linear.py

@@ -239,18 +239,12 @@ class RLinear(nn.Module):
                 #print(F"new Lookkback : {x.shape[1]}")
             
             self.transform_model(x.shape[1],1)
-            seq_last   = x[:,-1:].detach() 
-            x          = x - seq_last
-            x = F.linear(x, self.zero_shot_Linear)
-            #x = self.Linear(x.permute(0,2,1)).permute(0,2,1)
-            return x + seq_last
-
             #x = x * (x.shape[1]/512)
-            #x = self.revin_layer(x, 'norm')
-            #x = F.linear(x, self.zero_shot_Linear)
-            #x = self.revin_layer(x, 'denorm')
+            x = self.revin_layer(x, 'norm')
+            x = F.linear(x, self.zero_shot_Linear)
+            x = self.revin_layer(x, 'denorm')
             #x = x * (512/x.shape[1])
-            #return x
+            return x
 
                 
         if len(x_shape) == 2: