From 6927e0bb89f0c76dbbf5d14716cebd72ee13b2a5 Mon Sep 17 00:00:00 2001
From: Enno Hermann <enno.hermann@idiap.ch>
Date: Fri, 29 Nov 2024 16:17:02 +0100
Subject: [PATCH] fix(api): clearer error message when model doesn't support VC

---
 TTS/api.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/TTS/api.py b/TTS/api.py
index 12e82af5..ed828250 100644
--- a/TTS/api.py
+++ b/TTS/api.py
@@ -357,15 +357,17 @@ class TTS(nn.Module):
             target_wav (str):`
                 Path to the target wav file.
         """
-        wav = self.voice_converter.voice_conversion(source_wav=source_wav, target_wav=target_wav)
-        return wav
+        if self.voice_converter is None:
+            msg = "The selected model does not support voice conversion."
+            raise RuntimeError(msg)
+        return self.voice_converter.voice_conversion(source_wav=source_wav, target_wav=target_wav)
 
     def voice_conversion_to_file(
         self,
         source_wav: str,
         target_wav: str,
         file_path: str = "output.wav",
-    ):
+    ) -> str:
         """Voice conversion with FreeVC. Convert source wav to target speaker.
 
         Args: