Merge pull request IAHispano#751 from ShiromiyaG/oops

Fix speaker ID when it doesn't exist in the file
AznamirWoW · Sep 28, 2024 · c67d0a3 · c67d0a3
2 parents eeea1be + 01d73ba
commit c67d0a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py
@@ -309,7 +309,10 @@ def get_speakers_id(model):
     if model:
         model_data = torch.load(model, map_location="cpu")
         speakers_id = model_data.get("speakers_id", 0)
-        return list(range(speakers_id))
+        if speakers_id:
+            return list(range(speakers_id))
+        else:
+            return [0]
 
 
 # Inference tab

diff --git a/tabs/tts/tts.py b/tabs/tts/tts.py
@@ -147,7 +147,10 @@ def get_speakers_id(model):
     if model:
         model_data = torch.load(model, map_location="cpu")
         speakers_id = model_data.get("speakers_id", 0)
-        return list(range(speakers_id))
+        if speakers_id:
+            return list(range(speakers_id))
+        else:
+            return [0]
 
 
 # TTS tab