Browse Source

Fix speaker name bug

Leng Yue 2 years ago
parent
commit
f8e1a34d06
1 changed files with 1 additions and 1 deletions
  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(
             tokens, labels = self.pack_sentences(
                 final_text,
                 final_text,
                 semantics=final_semantic,
                 semantics=final_semantic,
-                speaker=None if self.use_speaker else response.name,
+                speaker=response.name if self.use_speaker else None,
                 add_bos=True,
                 add_bos=True,
             )
             )
             all_tokens.append(tokens)
             all_tokens.append(tokens)