Metadata-Version: 2.4
Name: stem-splitter
Version: 0.0.6
Summary: Simple, readable audio stem separation library
Author: anime-song
License-Expression: MIT
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: numpy>=1.24.1
Requires-Dist: tqdm>=4.66
Requires-Dist: librosa
Requires-Dist: einops>=0.7
Requires-Dist: torch>=2.7.0
Requires-Dist: torchaudio>=2.7.0
Requires-Dist: torchcodec

# 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待ちが発生します。生成後のカーネルはキャッシュされます。

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

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