Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -604,7 +604,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 604 |
if target_len < 512:
|
| 605 |
upsample = upsample.permute(0, 2, 1)
|
| 606 |
else:
|
| 607 |
-
upsample = upsample.permute(0, 2, 1)
|
| 608 |
|
| 609 |
# If input was 2D, remove the channel dimension
|
| 610 |
if x.shape[1] == 1:
|
|
|
|
| 604 |
if target_len < 512:
|
| 605 |
upsample = upsample.permute(0, 2, 1)
|
| 606 |
else:
|
| 607 |
+
upsample = upsample.permute(0, 2, 1)
|
| 608 |
|
| 609 |
# If input was 2D, remove the channel dimension
|
| 610 |
if x.shape[1] == 1:
|