Skip to main content

stem-splitter

BS-RoFormerを使用して音声を6ステム(bass、drums、other、vocals、guitar、piano)に分離するPythonパッケージです。

CUDA推論ではFP16 mixed precisionとTransformer層のtorch.compileを既定で使用します。compileを使用できない環境では警告を出してeager実行へフォールバックします。Windowsでcompileを使用する場合は、PyTorchのバージョンに対応するtriton-windowsを別途インストールしてください。

初回推論ではGPU向けカーネルを生成するため、数十秒程度のcompile待ちが発生します。生成後のカーネルはキャッシュされます。

pip install stem-splitter
stem-splitter input.wav --out-dir output

compileを明示的に無効化する場合は--no-compile、CUDAでFP32推論を行う場合は--full-precisionを指定します。

未ラベル蒸留の検証

teacher pseudo-labelによる小型BS-RoFormerの蒸留・弱stemアブレーションは、ライブラリ本体を変更せず distillation/ に分離しています。実測結果と採否判断は distillation/ABLATION_REPORT.md を参照してください。

Metadata

Release files for stem-splitter 0.0.9

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for stem-splitter 0.0.9
File Size Uploaded
stem_splitter-0.0.9.tar.gz 23.4 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for stem-splitter 0.0.9
File Interpreter ABI Platform
stem_splitter-0.0.9-py3-none-any.whl Python 3 none any Details

Total release size: 45.1 kB

Release files / stem_splitter-0.0.9.tar.gz

Download URL stem_splitter-0.0.9.tar.gz
Size 23.4 kB
Tags Source
SHA-256 checksum
How to use checksums
07b4828344055896c26b09b49e3a5eefb50dff0bff077f7313a733e1a8a09fed
BLAKE2b-256 checksum
How to use checksums
19b018a3012195a4d99ec8e151484123fe427d435ef844197d87572e68f0277e
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.10.6

Release files / stem_splitter-0.0.9-py3-none-any.whl

Download URL stem_splitter-0.0.9-py3-none-any.whl
Size 21.7 kB
Tags Python 3
SHA-256 checksum
How to use checksums
4a06b5453cc18d7880f3dcf4199407fd13e196e0039320111ce22fcda9839671
BLAKE2b-256 checksum
How to use checksums
1d60e442f6d1ec7340e6a2364bd28323ecec33079aced425cd88f7b767c79b16
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.10.6

Release history Release notifications | RSS feed

This release

0.0.9 This release

2 release files

0.0.8

2 release files

0.0.7

2 release files

0.0.6

2 release files

0.0.5

2 release files

0.0.4

2 release files

0.0.3

2 release files

0.0.2

2 release files

0.0.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page