Fix lang_seq inference error

yqzhishen · yqzhishen · commit fb6dbec666b1 · 2026-01-08T16:30:36.000+08:00
diff --git a/inference/ds_acoustic.py b/inference/ds_acoustic.py
@@ -88,7 +88,7 @@ def preprocess_input(self, param, idx=0):
             languages = torch.LongTensor([
                 (
                     self.lang_map[lang if '/' not in p else p.split('/', maxsplit=1)[0]]
-                    if self.phoneme_dictionary.is_cross_lingual(p)
+                    if self.phoneme_dictionary.is_cross_lingual(p if '/' in p else f'{lang}/{p}')
                     else 0
                 )
                 for p in param['ph_seq'].split()
diff --git a/inference/ds_variance.py b/inference/ds_variance.py
@@ -112,7 +112,7 @@ def preprocess_input(
             languages = torch.LongTensor([
                 (
                     self.lang_map[lang if '/' not in p else p.split('/', maxsplit=1)[0]]
-                    if self.phoneme_dictionary.is_cross_lingual(p)
+                    if self.phoneme_dictionary.is_cross_lingual(p if '/' in p else f'{lang}/{p}')
                     else 0
                 )
                 for p in param['ph_seq'].split()

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ def preprocess_input(self, param, idx=0):`
`88`	`88`	`languages = torch.LongTensor([`
`89`	`89`	`(`
`90`	`90`	`self.lang_map[lang if '/' not in p else p.split('/', maxsplit=1)[0]]`
`91`		`- if self.phoneme_dictionary.is_cross_lingual(p)`
	`91`	`+ if self.phoneme_dictionary.is_cross_lingual(p if '/' in p else f'{lang}/{p}')`
`92`	`92`	`else 0`
`93`	`93`	`)`
`94`	`94`	`for p in param['ph_seq'].split()`
Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def preprocess_input(`
`112`	`112`	`languages = torch.LongTensor([`
`113`	`113`	`(`
`114`	`114`	`self.lang_map[lang if '/' not in p else p.split('/', maxsplit=1)[0]]`
`115`		`- if self.phoneme_dictionary.is_cross_lingual(p)`
	`115`	`+ if self.phoneme_dictionary.is_cross_lingual(p if '/' in p else f'{lang}/{p}')`
`116`	`116`	`else 0`
`117`	`117`	`)`
`118`	`118`	`for p in param['ph_seq'].split()`