nodes/smthemex-ComfyUI_StableAvatar
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update vocal_seperator.py

Browse Source
This commit is contained in:
smthemex
2025-08-18 08:24:48 +08:00
committed by GitHub
parent f054915452
commit 68aa85c3d7
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vocal_seperator.py
+4 -3
View File
@@ -1,13 +1,14 @@
import argparse
# import argparse
import os
import shutil
import subprocess
from audio_separator.separator import Separator
def separator_audio(audio_file_path, audio_separator_model_file,cache_dir,audio_file_prefix):
from audio_separator.separator import Separator
#cache_dir = os.path.join(os.path.dirname(audio_file_path), "vocals")
os.makedirs(cache_dir, exist_ok=True)
audio_separator = Separator(
@@ -16,7 +17,7 @@ def separator_audio(audio_file_path, audio_separator_model_file,cache_dir,audio_
model_file_dir=os.path.dirname(audio_separator_model_file),
)
audio_separator.load_model(os.path.basename(audio_separator_model_file))
assert audio_separator.model_instance is not None, "Fail to load audio separate model."
# assert audio_separator.model_instance is not None, "Fail to load audio separate model."
outputs = audio_separator.separate(audio_file_path)
subfolder_path = os.path.dirname(audio_file_path)
vocal_audio_file = os.path.join(audio_separator.output_dir, outputs[0])