razmars commited on
Commit
47e81b4
·
verified ·
1 Parent(s): 8eeb171

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -224,7 +224,7 @@ class RLinear(nn.Module):
224
  if L < self.seq_len:
225
  x = x.clone()
226
  x = self.revin_layer(x, 'norm')
227
- x = F.linear(x, self.zero_shot_Linear).unsqueeze_(-1)
228
  x = self.revin_layer(x, 'denorm')
229
 
230
  else:
 
224
  if L < self.seq_len:
225
  x = x.clone()
226
  x = self.revin_layer(x, 'norm')
227
+ x = F.linear(x, self.zero_shot_Linear)
228
  x = self.revin_layer(x, 'denorm')
229
 
230
  else: