razmars commited on
Commit
294a5a0
·
verified ·
1 Parent(s): b3e1fe6

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +2 -2
modeling_super_linear.py CHANGED
@@ -238,9 +238,9 @@ class RLinear(nn.Module):
238
  # x: [Batch, Input length,Channel]
239
  x_shape = x.shape
240
  if x.shape[1] < self.seq_len:
241
- if self.zero_shot_Linear is None:
242
  #print(F"new Lookkback : {x.shape[1]}")
243
- self.transform_model(x.shape[1],2)
244
 
245
  x = x.clone()
246
  #x = x * (x.shape[1]/512)
 
238
  # x: [Batch, Input length,Channel]
239
  x_shape = x.shape
240
  if x.shape[1] < self.seq_len:
241
+ #if self.zero_shot_Linear is None:
242
  #print(F"new Lookkback : {x.shape[1]}")
243
+ self.transform_model(x.shape[1],2)
244
 
245
  x = x.clone()
246
  #x = x * (x.shape[1]/512)