fix test sentence synthesis

2021-11-02 16:33:40 +01:00 · 2021-11-02 16:33:40 +01:00 · 4d721bcabd
parent 0804806727
commit 4d721bcabd
1 changed files with 1 additions and 0 deletions
--- a/TTS/tts/utils/synthesis.py
+++ b/TTS/tts/utils/synthesis.py
@ -175,6 +175,7 @@ def embedding_to_torch(d_vector, cuda=False):
    if d_vector is not None:
        d_vector = np.asarray(d_vector)
        d_vector = torch.from_numpy(d_vector).type(torch.FloatTensor)
+        d_vector = d_vector.squeeze().unsqueeze(0)
    if cuda:
        return d_vector.cuda()
    return d_vector