razmars commited on
Commit
24e965a
·
verified ·
1 Parent(s): 099ba90

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -536,7 +536,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
536
  # ------------------------------------------------------------------
537
 
538
 
539
- def upsample_dim1(self, x, target_len: int = 512, mode: str = "linear"):
540
  # -------- bring the dim-1 axis to the PyTorch 1-D “length” position --------
541
  orig_shape = x.shape
542
  ndim = x.ndim
 
536
  # ------------------------------------------------------------------
537
 
538
 
539
+ def upsample_dim1(self, x, target_len: int = 512, mode: str = "bicubic"):
540
  # -------- bring the dim-1 axis to the PyTorch 1-D “length” position --------
541
  orig_shape = x.shape
542
  ndim = x.ndim