razmars commited on
Commit
7eddf81
·
verified ·
1 Parent(s): b5db567

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -215,7 +215,7 @@ class RLinear(nn.Module):
215
  x_shape = x.shape
216
  if x.shape[1] < self.seq_len:
217
  if self.zero_shot_Linear is None:
218
- self.transform_model(L)
219
  x = x.clone()
220
  x = self.revin_layer(x, 'norm')
221
  x = F.linear(x, self.zero_shot_Linear)
 
215
  x_shape = x.shape
216
  if x.shape[1] < self.seq_len:
217
  if self.zero_shot_Linear is None:
218
+ self.transform_model(x.shape[1])
219
  x = x.clone()
220
  x = self.revin_layer(x, 'norm')
221
  x = F.linear(x, self.zero_shot_Linear)