Rename duplicate UVR parameter in query parameters

jhj0517 · Nov 18, 2024 · bf709ac · bf709ac
1 parent 382d0a7
commit bf709ac
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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,