Update modeling_super_linear.py

modeling_super_linear.py

@@ -239,10 +239,9 @@ class RLinear(nn.Module):
                 #print(F"new Lookkback : {x.shape[1]}")
             
             self.transform_model(x.shape[1],1)
-            seq_last   = x[:,-1:,:].detach() 
+            seq_last   = x[:,-1:].detach() 
             x          = x - seq_last
-            x          = x.squeeze(-1)
-            x = F.linear(x, self.zero_shot_Linear).unsqueeze_(-1)
+            x = F.linear(x, self.zero_shot_Linear)
             #x = self.Linear(x.permute(0,2,1)).permute(0,2,1)
             return x + seq_last