update wan2.2-S2V training

2026-03-22 08:40:47 +00:00 · 2025-10-21 19:59:44 +08:00
parent b168d7aa8b
commit 30292d9411
5 changed files with 7 additions and 13 deletions
--- a/diffsynth/trainers/unified_dataset.py
+++ b/diffsynth/trainers/unified_dataset.py
@@ -231,7 +231,7 @@ class LoadAudio(DataProcessingOperator):
    def __call__(self, data: str):
        import librosa
        input_audio, sample_rate = librosa.load(data, sr=self.sr)
-        return {'input_audio':input_audio, 'sample_rate':sample_rate}
+        return input_audio


 class UnifiedDataset(torch.utils.data.Dataset):