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

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +3 -3
modeling_super_linear.py CHANGED
@@ -216,10 +216,10 @@ class RLinear(nn.Module):
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)
222
- x = self.revin_layer(x, 'denorm')
223
  return x
224
 
225
 
 
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)
222
+ #x = self.revin_layer(x, 'denorm')
223
  return x
224
 
225