Update modeling_super_linear.py

modeling_super_linear.py

@@ -206,7 +206,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
+            final_scaling  = 1
             new_W          = new_W * final_scaling
             self.zero_shot_Linear        = new_W
 
@@ -640,7 +640,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
             x_enc = self.revin_layer(x_enc, 'norm')
             x_enc = self.fourier_interp_dim1(x_enc)
 
-        self.backbone.inf_pred_len = 336
+        self.backbone.inf_pred_len = 192
 
         # backbone returns (B, pred_len, C)