Update modeling_super_linear.py

modeling_super_linear.py

@@ -210,8 +210,8 @@ class RLinear(nn.Module):
             new_W          = new_W * final_scaling
 
             self.zero_shot_Linear        = new_W
-        else:
-            W              = self.Linear.weight.detach()
+        elif mode ==2:
+            W = self.Linear.weight.detach()
             W4d = W.unsqueeze(0).unsqueeze(0)            # (1, 1, out, in)
 
             # resize   H → self.horizon   and   W → new_lookback
@@ -223,6 +223,22 @@ class RLinear(nn.Module):
                     )[0, 0]                                 # drop the two singleton dims
 
             self.zero_shot_Linear = new_W                  # shape (self.horizon, new_lookback)
+        else:
+            W   = self.Linear.weight.detach()
+            W   = W[:, -new_lookback:] 
+            W4d = W.unsqueeze(0).unsqueeze(0)            # (1, 1, out, in)
+
+            # resize   H → self.horizon   and   W → new_lookback
+            new_W = F.interpolate(
+                        W4d,
+                        size=(self.seq_len , new_lookback),   # (H_out, W_out)
+                        mode='bilinear',
+                        align_corners=False
+                    )[0, 0]                                 # drop the two singleton dims
+
+            W_now = torch.cat((W, new_W), dim=1)  
+            self.zero_shot_Linear = new_W                  
+