mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix mono input

Browse Source
This commit is contained in:
Gong Junmin
2025-05-10 21:37:38 +08:00
parent 7da27aa311
commit b86a71088c
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/music_dcae/music_dcae_pipeline.py
+2
View File
@@ -61,6 +61,8 @@ class MusicDCAE(ModelMixin, ConfigMixin, FromOriginalModelMixin):
def load_audio(self, audio_path): def load_audio(self, audio_path):
audio, sr = torchaudio.load(audio_path) audio, sr = torchaudio.load(audio_path)
if audio.shape[0] == 1:
audio = audio.repeat(2, 1)
return audio, sr return audio, sr
def forward_mel(self, audios): def forward_mel(self, audios):