Update modeling_super_linear.py

modeling_super_linear.py

@@ -205,6 +205,7 @@ class RLinear(nn.Module):
         original_norm  = torch.norm(W, p=2)
         new_norm       = torch.norm(new_W, p=2)
         final_scaling  = original_norm / new_norm if new_norm.item() != 0 else 1.0
+        final_scaling  = 1
         new_W          = new_W * final_scaling
 
         self.zero_shot_Linear        = new_W
@@ -216,10 +217,10 @@ class RLinear(nn.Module):
         if x.shape[1] < self.seq_len:
             if self.zero_shot_Linear is None:
                 self.transform_model(x.shape[1])
-            #x = x.clone()
-            #x = self.revin_layer(x, 'norm')
+            x = x.clone()
+            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, 'denorm')
             return x