razmars commited on
Commit
2bf5e6c
·
verified ·
1 Parent(s): 0be4f34

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +0 -1
modeling_super_linear.py CHANGED
@@ -567,7 +567,6 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
567
  def fourier_interp_dim1(self,x, target_len: int = 512):
568
 
569
  L = x.size(1)
570
- assert L == 48, "dim-1 length must be 48"
571
 
572
  X = torch.fft.rfft(x, dim=1) # (..., 25, ...)
573
  pad = target_len // 2 + 1 - X.size(1)
 
567
  def fourier_interp_dim1(self,x, target_len: int = 512):
568
 
569
  L = x.size(1)
 
570
 
571
  X = torch.fft.rfft(x, dim=1) # (..., 25, ...)
572
  pad = target_len // 2 + 1 - X.size(1)