Added support for UVR-MDX-NET-Inst_HQ_5.onnx model

audio_separator/model-data.json

@@ -10,5 +10,13 @@
             "is_bv_model_rebalanced": 0.0
         }
     },
-    "mdx_model_data": {}
-}
+    "mdx_model_data": {
+        "cb790d0c913647ced70fc6b38f5bea1a": {
+            "compensate": 1.010,
+            "mdx_dim_f_set": 2560,
+            "mdx_dim_t_set": 8,
+            "mdx_n_fft_scale_set": 5120,
+            "primary_stem": "Instrumental"
+        }
+    }
+}

audio_separator/models.json

@@ -2,10 +2,16 @@
     "vr_download_list": {
         "VR Arch Single Model v4: UVR-De-Reverb by aufr33-jarredou": "UVR-De-Reverb-aufr33-jarredou.pth"
     },
-    "mdx_download_list": {},
+    "mdx_download_list": {
+        "MDX-Net Model: UVR-MDX-NET Inst HQ 5": "UVR-MDX-NET-Inst_HQ_5.onnx"
+    },
     "mdx23c_download_list": {
-        "MDX23C Model: MDX23C De-Reverb by aufr33-jarredou": {"MDX23C-De-Reverb-aufr33-jarredou.ckpt":"config_dereverb_mdx23c.yaml"},
-        "MDX23C Model: MDX23C DrumSep by aufr33-jarredou": {"MDX23C-DrumSep-aufr33-jarredou.ckpt":"config_drumsep_mdx23c.yaml"}
+        "MDX23C Model: MDX23C De-Reverb by aufr33-jarredou": {
+            "MDX23C-De-Reverb-aufr33-jarredou.ckpt": "config_dereverb_mdx23c.yaml"
+        },
+        "MDX23C Model: MDX23C DrumSep by aufr33-jarredou": {
+            "MDX23C-DrumSep-aufr33-jarredou.ckpt": "config_drumsep_mdx23c.yaml"
+        }
     },
     "roformer_download_list": {
         "Roformer Model: Mel-Roformer-Karaoke-Aufr33-Viperx": {
@@ -78,4 +84,4 @@
             "mel_band_roformer_bleed_suppressor_v1.ckpt": "config_mel_band_roformer_bleed_suppressor_v1.yaml"
         }
     }
-}
+}

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "audio-separator"
-version = "0.28.4"
+version = "0.28.5"
 description = "Easy to use audio stem separation, using various models from UVR trained primarily by @Anjok07"
 authors = ["Andrew Beveridge <[email protected]>"]
 license = "MIT"