Update modeling_super_linear.py

modeling_super_linear.py

@@ -424,7 +424,7 @@ class Model(nn.Module):
             outputs = [out]
             ar_x = torch.cat([x, out], dim=1)[:, -self.train_seq_len:]
             for i in range(0, pred_len, self.train_pred_len):
-                ar_out, _ = self.moe(ar_x)
+                ar_out = self.moe(ar_x)
                 outputs.append(ar_out)
                 ar_x = torch.cat([ar_x, ar_out], dim=1)[:, -self.train_seq_len:]
             out = torch.cat(outputs, dim=1)[:, :pred_len]