Tell the hifigan type of a vocoder model by searching full text

3 years ago · 7f799d322f
parent a1f2e4a790
commit 7f799d322f
1 changed files with 1 additions and 1 deletions
--- a/toolbox/init.py
+++ b/toolbox/init.py
@ -447,7 +447,7 @@ class Toolbox:
            return 
        # Sekect vocoder based on model name
        model_config_fpath = None
-        if model_fpath.name[0] == "g":
+        if model_fpath.name is not None and model_fpath.name.find("hifigan") > -1:
            vocoder = gan_vocoder
            self.ui.log("set hifigan as vocoder")
            # search a config file