mei/SoulX-Singer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update example code for each preprocess module

Browse Source
This commit is contained in:
jlqian98
2026-03-04 15:02:34 +08:00
parent 79e1f06d7d
commit 7bc2dfea6e
5 changed files with 36 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
preprocess/tools/vocal_separation/model.py
+5 -5
View File
@@ -214,12 +214,12 @@ class VocalSeparator:
if __name__ == "__main__":
m = VocalSeparator(
sep_model_path="pretrained_models/mel-band-roformer-karaoke/mel_band_roformer_karaoke_becruily.ckpt",
sep_config_path="pretrained_models/mel-band-roformer-karaoke/config_karaoke_becruily.yaml",
der_model_path="pretrained_models/dereverb_mel_band_roformer/dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt",
der_config_path="pretrained_models/dereverb_mel_band_roformer/dereverb_mel_band_roformer_anvuew.yaml",
sep_model_path="pretrained_models/SoulX-Singer-Preprocess/mel-band-roformer-karaoke/mel_band_roformer_karaoke_becruily.ckpt",
sep_config_path="pretrained_models/SoulX-Singer-Preprocess/mel-band-roformer-karaoke/config_karaoke_becruily.yaml",
der_model_path="pretrained_models/SoulX-Singer-Preprocess/dereverb_mel_band_roformer/dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt",
der_config_path="pretrained_models/SoulX-Singer-Preprocess/dereverb_mel_band_roformer/dereverb_mel_band_roformer_anvuew.yaml",
device="cuda"
)
out = m.process("example/test/separation_test.mp3")
out = m.process("example/audio/zh_prompt.mp3")
print(out.vocals_dereverbed.shape)