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.8

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.8
File Size Uploaded
stem_splitter-0.0.8.tar.gz 22.9 kB Details

Built distribution (wheel)

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

Total release size: 44.3 kB

Release files / stem_splitter-0.0.8.tar.gz

Download URL stem_splitter-0.0.8.tar.gz
Size 22.9 kB
Tags Source
SHA-256 checksum
How to use checksums
a1c37d8d69719118c3ec3c6b2d95ba106ef50185021b2fbf9468adbfc91f79de
BLAKE2b-256 checksum
How to use checksums
65d3ad1996a30a1b08ccec1e2884f3e26f7d260a75aa9e32d5a8c8782f400ad3
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.8-py3-none-any.whl

Download URL stem_splitter-0.0.8-py3-none-any.whl
Size 21.4 kB
Tags Python 3
SHA-256 checksum
How to use checksums
d843e47ea4d15e431fb6cf80f3cdd534a2ebc16b4ac1a6672732bff6490200a9
BLAKE2b-256 checksum
How to use checksums
13d1e865fc89f11db4fcd8bd3d2c840d2450d72d6a9c5ff616966ea7ac9d49ff
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

0.0.9

2 release files

This release

0.0.8 This release

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