Guides
GPU acceleration on macOS
This feature is not supported yet!
There is some ongoing work to support Metal Performance Shaders in PyTorch.
In the future, you will be able to use the -d
or --device
option in Stemgen to use Metal Performance Shaders.
python3 stemgen.py track.wav -d mps