Update modeling_super_linear.py
Browse files- 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 = "
|
| 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
|