Skip to content

Commit

Permalink
2 new models and minor fix to README (#165)
Browse files Browse the repository at this point in the history
* new models

* list upcoming MDX23C models
  • Loading branch information
Eddycrack864 authored Dec 12, 2024
1 parent ba81920 commit e598b49
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -348,7 +348,7 @@ You can also rename specific stems:
- **`output_dir`:** (Optional) Directory where the separated files will be saved. If not specified, uses the current directory.
- **`output_format`:** (Optional) Format to encode output files, any common format (WAV, MP3, FLAC, M4A, etc.). `Default: WAV`
- **`normalization_threshold`:** (Optional) The amount by which the amplitude of the output audio will be multiplied. `Default: 0.9`
- **`amplification_threshold`:** (Optional) The minimum amplitude level at which the waveform will be amplified. If the peak amplitude of the audio is below this threshold, the waveform will be scaled up to meet it. `Default: 0.6`
- **`amplification_threshold`:** (Optional) The minimum amplitude level at which the waveform will be amplified. If the peak amplitude of the audio is below this threshold, the waveform will be scaled up to meet it. `Default: 0.0`
- **`output_single_stem`:** (Optional) Output only a single stem, such as 'Instrumental' and 'Vocals'. `Default: None`
- **`invert_using_spec`:** (Optional) Flag to invert using spectrogram. `Default: False`
- **`sample_rate`:** (Optional) Set the sample rate of the output audio. `Default: 44100`
Expand Down
6 changes: 6 additions & 0 deletions audio_separator/models.json
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,12 @@
},
"Roformer Model: BS-Roformer-De-Reverb": {
"deverb_bs_roformer_8_384dim_10depth.ckpt": "deverb_bs_roformer_8_384dim_10depth_config.yaml"
},
"Roformer Model: Mel-Roformer-Vocals-Kim": {
"vocals_mel_band_roformer.ckpt": "vocals_mel_band_roformer.yaml"
},
"Roformer Model: MelBand Roformer Kim | Inst V1 (E) by Unwa": {
"melband_roformer_inst_v1e.ckpt": "config_melbandroformer_inst.yaml"
}
}
}
1 change: 1 addition & 0 deletions audio_separator/separator/separator.py
Original file line number Diff line number Diff line change
Expand Up @@ -420,6 +420,7 @@ def list_supported_model_files(self):
**model_downloads_list["mdx23c_download_list"],
**model_downloads_list["mdx23c_download_vip_list"],
**model_downloads_list["roformer_download_list"],
**audio_separator_models_list["mdx23c_download_list"],
**audio_separator_models_list["roformer_download_list"],
},
}
Expand Down

0 comments on commit e598b49

Please sign in to comment.