fix test sentence synthesis

2026-07-11 04:51:29 +02:00 · 2021-11-02 16:33:40 +01:00
parent 20ac31dc71
commit 88d6399e12
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