Spaces:

jhansss
/

SingingSDS

Running

jhansss commited on Jul 3

Commit

d5992a5

1 Parent(s): 2195601

fix espnet mixed svs

Files changed (1) hide show

modules/svs/espnet.py CHANGED Viewed

@@ -107,8 +107,8 @@ class ESPNetSVS(AbstractSVSModel):
             output_dict = self.model(batch, sids=sid)
         elif self.model_id == "espnet/mixdata_svs_visinger2_spkemb_lang_pretrained":
             langs = {
-                "zh": 2,
-                "jp": 1,
             }
             if language not in langs:
                 raise ValueError(

             output_dict = self.model(batch, sids=sid)
         elif self.model_id == "espnet/mixdata_svs_visinger2_spkemb_lang_pretrained":
             langs = {
+                "mandarin": 2,
+                "japanese": 1,
             }
             if language not in langs:
                 raise ValueError(