diff --git a/backend/bgm_separation/router.py b/backend/bgm_separation/router.py index 0ffc182..ea4f396 100644 --- a/backend/bgm_separation/router.py +++ b/backend/bgm_separation/router.py @@ -50,8 +50,8 @@ async def run_bgm_separation( start_time = datetime.utcnow() instrumental, vocal, filepaths = get_bgm_separation_inferencer().separate( audio=audio, - model_name=params.model_size, - device=params.device, + model_name=params.uvr_model_size, + device=params.uvr_device, segment_size=params.segment_size, save_file=False, progress=gr.Progress() diff --git a/modules/whisper/base_transcription_pipeline.py b/modules/whisper/base_transcription_pipeline.py index c7d237a..59c5727 100644 --- a/modules/whisper/base_transcription_pipeline.py +++ b/modules/whisper/base_transcription_pipeline.py @@ -110,8 +110,8 @@ def run(self, if bgm_params.is_separate_bgm: music, audio, _ = self.music_separator.separate( audio=audio, - model_name=bgm_params.model_size, - device=bgm_params.device, + model_name=bgm_params.uvr_model_size, + device=bgm_params.uvr_device, segment_size=bgm_params.segment_size, save_file=bgm_params.save_file, progress=progress diff --git a/modules/whisper/data_classes.py b/modules/whisper/data_classes.py index 0d3ab22..b00a4c9 100644 --- a/modules/whisper/data_classes.py +++ b/modules/whisper/data_classes.py @@ -185,11 +185,11 @@ def to_gradio_inputs(cls, class BGMSeparationParams(BaseParams): """Background music separation parameters""" is_separate_bgm: bool = Field(default=False, description="Enable background music separation") - model_size: str = Field( + uvr_model_size: str = Field( default="UVR-MDX-NET-Inst_HQ_4", description="UVR model size" ) - device: str = Field(default="cuda", description="Device to run UVR model.") + uvr_device: str = Field(default="cuda", description="Device to run UVR model.") segment_size: int = Field( default=256, gt=0,