Leng Yue 2 лет назад
Родитель
Сommit
f8e1a34d06
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      fish_speech/datasets/text.py

+ 1 - 1
fish_speech/datasets/text.py

@@ -387,7 +387,7 @@ class AutoAugTextDataset(IterableDataset):
             tokens, labels = self.pack_sentences(
                 final_text,
                 semantics=final_semantic,
-                speaker=None if self.use_speaker else response.name,
+                speaker=response.name if self.use_speaker else None,
                 add_bos=True,
             )
             all_tokens.append(tokens)