Update modeling_super_linear.py

modeling_super_linear.py

@@ -419,9 +419,10 @@ class Model(nn.Module):
             out, expert_probs = self.moe(x, get_prob=True)
         else:
             out = self.moe(x)
-        print("test!")
+
+        print(pred_len)
+        print(self.train_pred_len)
         if self.train_pred_len < pred_len:
-            print("test2")
             outputs = [out]
             ar_x = torch.cat([x, out], dim=1)[:, -self.seq_len:]
             for i in range(0, inf_pred_len, self.train_pred_len):