Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -613,7 +613,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 613 |
# 4. Renormalise amplitudes:
|
| 614 |
# irfft divides by `target_len`, whereas the forward rfft used length `L`.
|
| 615 |
# Multiply by (target_len / L) so DC and low-freq amplitudes match input.
|
| 616 |
-
|
| 617 |
|
| 618 |
return y
|
| 619 |
|
|
|
|
| 613 |
# 4. Renormalise amplitudes:
|
| 614 |
# irfft divides by `target_len`, whereas the forward rfft used length `L`.
|
| 615 |
# Multiply by (target_len / L) so DC and low-freq amplitudes match input.
|
| 616 |
+
y *= target_len / L
|
| 617 |
|
| 618 |
return y
|
| 619 |
|