From ba6e56e01c8fd1a42177fc5717e29fa0d6990e05 Mon Sep 17 00:00:00 2001
From: Edresson Casanova <edresson1@gmail.com>
Date: Fri, 18 Feb 2022 19:25:29 +0000
Subject: [PATCH] Fix Glow-TTS multi-speaker inference

---
 TTS/tts/models/glow_tts.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/TTS/tts/models/glow_tts.py b/TTS/tts/models/glow_tts.py
index 7dbfdd09..8f3b3804 100644
--- a/TTS/tts/models/glow_tts.py
+++ b/TTS/tts/models/glow_tts.py
@@ -170,6 +170,8 @@ class GlowTTS(BaseTTS):
         if g is not None:
             if hasattr(self, "emb_g"):
                 # use speaker embedding layer
+                if not g.size(): # if is a scalar
+                    g = g.unsqueeze(0) # unsqueeze
                 g = F.normalize(self.emb_g(g)).unsqueeze(-1)  # [b, h, 1]
             else:
                 # use d-vector