Update modeling_super_linear.py

modeling_super_linear.py

@@ -219,14 +219,13 @@ class RLinear(nn.Module):
                 #print(F"new Lookkback : {x.shape[1]}")
                 self.transform_model(x.shape[1])
 
-            if len(x_shape) == 2:
-                x = x.unsqueeze(-1)
+
             x = x.clone()
+            x = x * (x.shape[1]/512)
             x = self.revin_layer(x, 'norm')
-            x = F.linear(x.permute(0,2,1), self.zero_shot_Linear).permute(0,2,1).clone()
+            x = F.linear(x, self.zero_shot_Linear)
             x = self.revin_layer(x, 'denorm')
-            if len(x_shape) == 2:
-                x = x.squeeze(-1)
+            x = x * (512/x.shape[1])
             return x