razmars commited on
Commit
9c69067
·
verified ·
1 Parent(s): 38cedcd

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -510,7 +510,7 @@ class superLinear(nn.Module):
510
  return cycle
511
 
512
 
513
- def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
514
 
515
  if inf_pred_len is None:
516
  inf_pred_len = self.inf_pred_len
 
510
  return cycle
511
 
512
 
513
+ 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):
514
 
515
  if inf_pred_len is None:
516
  inf_pred_len = self.inf_pred_len