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

delete irrelevant code

Browse Source
This commit is contained in:
chuxij
2025-04-24 23:59:34 +08:00
parent d6f5a2d911
commit f91ce4867f
10 changed files with 3 additions and 1290 deletions
Download Patch File Download Diff File Expand all files Collapse all files
music_dcae/music_dcae_pipeline.py
+3 -3
View File
@@ -135,7 +135,7 @@ class MusicDCAE(nn.Module):
if __name__ == "__main__":
audio, sr = torchaudio.load("/root/data/repo/gongjunmin/sag_train/orig2.wav")
audio, sr = torchaudio.load("test.wav")
audio_lengths = torch.tensor([audio.shape[1]])
audios = audio.unsqueeze(0)
@@ -151,5 +151,5 @@ if __name__ == "__main__":
print("latents shape: ", latents.shape)
print("latent_lengths: ", latent_lengths)
print("sr: ", sr)
torchaudio.save("/root/data/repo/gongjunmin/sag_train/reconstructed.wav", pred_wavs[0], sr)
print("reconstructed wav saved to /root/data/repo/gongjunmin/sag_train/reconstructed.wav")
torchaudio.save("test_reconstructed.flac", pred_wavs[0], sr)
print("/test_reconstructed.flac")