Update modeling_super_linear.py
Browse files- modeling_super_linear.py +2 -0
modeling_super_linear.py
CHANGED
|
@@ -327,6 +327,8 @@ class SparseNoisyMoE(nn.Module):
|
|
| 327 |
|
| 328 |
if inputs.dim() == 2:
|
| 329 |
I = I.squeeze(2)
|
|
|
|
|
|
|
| 330 |
|
| 331 |
|
| 332 |
def fourier_interp_dim1(self,x, target_len: int = 512):
|
|
|
|
| 327 |
|
| 328 |
if inputs.dim() == 2:
|
| 329 |
I = I.squeeze(2)
|
| 330 |
+
|
| 331 |
+
return I
|
| 332 |
|
| 333 |
|
| 334 |
def fourier_interp_dim1(self,x, target_len: int = 512):
|