Update modeling_super_linear.py

modeling_super_linear.py

@@ -239,11 +239,12 @@ 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
+            seq_last   = x[:,-1:,:].detach() 
+            x          = x - seq_last
+            x          = x.squeeze(-1)
+            x = F.linear(x, self.zero_shot_Linear).unsqueeze_(-1)
+            #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')