Update text2music_dataset.py

fix_speaker_emb_none
2025-05-08 18:33:04 +08:00
parent 2c0d5df515
commit 5f8c0bb4b5
1 changed files with 2 additions and 1 deletions
@@ -456,7 +456,8 @@ class Text2MusicDataset(Dataset):
        speaker_emb_path = item.get("speaker_emb_path")
        if not speaker_emb_path:
            speaker_emb = self.get_speaker_emb_file(speaker_emb_path)
-        else:
+
+        if speaker_emb is None:
            speaker_emb = torch.zeros(512)

        # Process prompt/tags