Update modeling_super_linear.py

modeling_super_linear.py

@@ -420,6 +420,15 @@ class Model(nn.Module):
         else:
             out = self.moe(x)
 
+        if self.max_horizon < inf_pred_len:
+            outputs = [out]
+            ar_x = torch.cat([x, out], dim=1)[:, -self.seq_len:]
+            for i in range(0, inf_pred_len, self.max_horizon):
+                ar_out, _ = self.moe(ar_x)
+                outputs.append(ar_out)
+                ar_x = torch.cat([ar_x, ar_out], dim=1)[:, -self.seq_len:]
+            out = torch.cat(outputs, dim=1)[:, :inf_pred_len]
+
         # Reshape back
         out = out.reshape(B, V, out.size(-1))