Update TTS.py

3 weeks ago · 4d8ebf8523
parent e476b01f30
commit 4d8ebf8523
1 changed files with 2 additions and 2 deletions
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@ -304,8 +304,8 @@ class TTS_Config:
            configs: dict = self._load_configs(self.configs_path)

        assert isinstance(configs, dict)
-        version = configs.get("version", "v2")
-        assert version in ["v1", "v2", "v3", "v4", "v2Pro", "v2ProPlus"]
+        version = configs.get("version", "v2").lower()
+        assert version in ["v1", "v2", "v3", "v4", "v2pro", "v2proplus"]
        self.default_configs[version] = configs.get(version, self.default_configs[version])
        self.configs: dict = configs.get("custom", deepcopy(self.default_configs[version]))