razmars commited on
Commit
03ea7d0
·
verified ·
1 Parent(s): 956cd48

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -218,7 +218,7 @@ class RLinear(nn.Module):
218
  if x.shape[1] < self.seq_len:
219
  if self.zero_shot_Linear is None:
220
  #print(F"new Lookkback : {x.shape[1]}")
221
- self.transform_model(x.shape[1])
222
 
223
  x = x.clone()
224
  #x = x * (x.shape[1]/512)
 
218
  if x.shape[1] < self.seq_len:
219
  if self.zero_shot_Linear is None:
220
  #print(F"new Lookkback : {x.shape[1]}")
221
+ self.transform_model(x.shape[1],2)
222
 
223
  x = x.clone()
224
  #x = x * (x.shape[1]/512)