From 7f799d322f7599d466c61a527e8c01ef117a4ed2 Mon Sep 17 00:00:00 2001
From: babysor00 <babysor00@gmail.com>
Date: Sat, 30 Apr 2022 10:31:01 +0800
Subject: [PATCH] Tell the hifigan type of a vocoder model by searching full
 text

---
 toolbox/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolbox/__init__.py b/toolbox/__init__.py
index d0cd20e..3d03397 100644
--- 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