razmars commited on
Commit
21831ca
·
verified ·
1 Parent(s): bc79359

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -219,7 +219,7 @@ class RLinear(nn.Module):
219
  self.transform_model(x.shape[1])
220
  x = x.clone()
221
  x = self.revin_layer(x, 'norm')
222
- x = F.linear(x, self.zero_shot_Linear)
223
  x = self.revin_layer(x, 'denorm')
224
  return x
225
 
 
219
  self.transform_model(x.shape[1])
220
  x = x.clone()
221
  x = self.revin_layer(x, 'norm')
222
+ x = F.linear(x, self.zero_shot_Linear).clone()
223
  x = self.revin_layer(x, 'denorm')
224
  return x
225