Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test(freevc): fix output length check

This commit is contained in:
Enno Hermann 2024-12-02 00:09:39 +01:00
parent 546f43cb25
commit 9ef2c7ed62
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/vc_tests/test_freevc.py
View File

@ -31,7 +31,7 @@ class TestFreeVC(unittest.TestCase):
@staticmethod @staticmethod
def _create_inputs_inference(): def _create_inputs_inference():
source_wav = torch.rand(16000) source_wav = torch.rand(15999)
target_wav = torch.rand(16000) target_wav = torch.rand(16000)
return source_wav, target_wav return source_wav, target_wav
@ -96,8 +96,8 @@ class TestFreeVC(unittest.TestCase):
source_wav, target_wav = self._create_inputs_inference() source_wav, target_wav = self._create_inputs_inference()
output_wav = model.voice_conversion(source_wav, target_wav) output_wav = model.voice_conversion(source_wav, target_wav)
assert ( assert (
output_wav.shape[0] + config.audio.hop_length == source_wav.shape[0] output_wav.shape[0] == source_wav.shape[0] - source_wav.shape[0] % config.audio.hop_length
), f"{output_wav.shape} != {source_wav.shape}" ), f"{output_wav.shape} != {source_wav.shape}, {config.audio.hop_length}"
def test_train_step(self): ... def test_train_step(self): ...