fix process_legacy_versions 1.1.0-transition

Artrajz · May 2, 2024 · 3c5d6b4 · 3c5d6b4
1 parent 0414037
commit 3c5d6b4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bert_vits2/bert_vits2.py b/bert_vits2/bert_vits2.py
@@ -52,9 +52,10 @@ def __init__(self, model_path, config, device=torch.device("cpu"), **kwargs):
             self.num_tones = num_tones_v111
             self.text_extra_str_map.update({"zh": "_v100"})
 
-        elif self.version in ["1.1.0-transition"]:
+        elif self.version in ["1.1.0_transition"]:
             """
             chinese-roberta-wwm-ext-large
+            bert-base-japanese-v3
             """
             self.version = "1.1.0-transition"
             self.hps_ms.model.n_layers_trans_flow = 3