Spaces:
Running
on
Zero
Running
on
Zero
File size: 401 Bytes
45ee559 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import numpy as np
import torch
from TTS.vocoder.models.melgan_generator import MelganGenerator
def test_melgan_generator():
model = MelganGenerator()
print(model)
dummy_input = torch.rand((4, 80, 64))
output = model(dummy_input)
assert np.all(output.shape == (4, 1, 64 * 256))
output = model.inference(dummy_input)
assert np.all(output.shape == (4, 1, (64 + 4) * 256))
|