razmars commited on
Commit
c3cf38d
·
verified ·
1 Parent(s): 786dee2

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -2
modeling_super_linear.py CHANGED
@@ -240,10 +240,9 @@ class RLinear(nn.Module):
240
 
241
  self.transform_model(x.shape[1],1)
242
 
243
- x = x.clone()
244
  #x = x * (x.shape[1]/512)
245
  x = self.revin_layer(x, 'norm')
246
- x = F.linear(x, self.zero_shot_Linear).clone()
247
  x = self.revin_layer(x, 'denorm')
248
  #x = x * (512/x.shape[1])
249
  return x
 
240
 
241
  self.transform_model(x.shape[1],1)
242
 
 
243
  #x = x * (x.shape[1]/512)
244
  x = self.revin_layer(x, 'norm')
245
+ x = F.linear(x, self.zero_shot_Linear)
246
  x = self.revin_layer(x, 'denorm')
247
  #x = x * (512/x.shape[1])
248
  return x