Merge pull request #116 from ace-step/support_mono_input

fix mono input
2025-05-10 21:38:05 +08:00
parent 7da27aa311 b86a71088c
commit 6eaf2589e3
1 changed files with 2 additions and 0 deletions
@@ -61,6 +61,8 @@ class MusicDCAE(ModelMixin, ConfigMixin, FromOriginalModelMixin):

    def load_audio(self, audio_path):
        audio, sr = torchaudio.load(audio_path)
+        if audio.shape[0] == 1:
+            audio = audio.repeat(2, 1)
        return audio, sr

    def forward_mel(self, audios):