Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -599,7 +599,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 599 |
x = x.unsqueeze(1) # [B, 1, L]
|
| 600 |
|
| 601 |
|
| 602 |
-
upsample =
|
| 603 |
|
| 604 |
# If input was 2D, remove the channel dimension
|
| 605 |
if size == 2:
|
|
|
|
| 599 |
x = x.unsqueeze(1) # [B, 1, L]
|
| 600 |
|
| 601 |
|
| 602 |
+
upsample = F.interpolate(x, size=target_len, mode='linear', align_corners=False)
|
| 603 |
|
| 604 |
# If input was 2D, remove the channel dimension
|
| 605 |
if size == 2:
|