Update modeling_super_linear.py

modeling_super_linear.py

@@ -239,13 +239,18 @@ class RLinear(nn.Module):
                 #print(F"new Lookkback : {x.shape[1]}")
             
             self.transform_model(x.shape[1],1)
+            new_x    = x.unsqueeze(-1)
+            seq_last = new_x[:,-1:,:].detach()
+            new_x    = new_x - seq_last
+            new_x    =  self.Linear(new_x.permute(0,2,1)).permute(0,2,1)
+            return new_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: