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

Merge pull request #116 from ace-step/support_mono_input

Browse Source 
fix mono input
This commit is contained in:
Gong Junmin authored and GitHub committed 2025-05-10 21:38:05 +08:00
2 parents 7da27aa311 b86a71088c
commit 6eaf2589e3
1 file changed
+2
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):
audio, sr = torchaudio.load(audio_path)
if audio.shape[0] == 1:
audio = audio.repeat(2, 1)
return audio, sr
def forward_mel(self, audios):