Skip to content

Commit ea49c55

Browse files
committed
Changed default model to 2_HP-UVR.pth
1 parent 861482b commit ea49c55

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

audio_separator/separator/separator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ def list_supported_model_files(self):
354354
}
355355
return model_files_grouped
356356

357-
def load_model(self, model_filename="UVR-MDX-NET-Inst_HQ_3.onnx"):
357+
def load_model(self, model_filename="2_HP-UVR.pth"):
358358
"""
359359
This method loads the separation model into memory, downloading it first if necessary.
360360
"""

audio_separator/utils/cli.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def main():
2424
parser.add_argument("--list_models", action="store_true", help="List all supported models and exit.")
2525

2626
parser.add_argument(
27-
"--model_filename", default="UVR-MDX-NET-Inst_HQ_3.onnx", help="Optional: model filename to be used for separation (default: %(default)s). Example: --model_filename=UVR_MDXNET_KARA_2.onnx"
27+
"--model_filename", default="2_HP-UVR.pth", help="Optional: model filename to be used for separation (default: %(default)s). Example: --model_filename=UVR_MDXNET_KARA_2.onnx"
2828
)
2929

3030
parser.add_argument("--model_file_dir", default="/tmp/audio-separator-models/", help="Optional: model files directory (default: %(default)s). Example: --model_file_dir=/app/models")

0 commit comments

Comments
 (0)