razmars commited on
Commit
7eb8c8e
·
verified ·
1 Parent(s): b027194

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -490,7 +490,7 @@ class superLinear(nn.Module):
490
  return cycle
491
 
492
 
493
- def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
494
 
495
  if inf_pred_len is None:
496
  inf_pred_len = self.inf_pred_len
 
490
  return cycle
491
 
492
 
493
+ def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False, inf_pred_len=None):
494
 
495
  if inf_pred_len is None:
496
  inf_pred_len = self.inf_pred_len