Thanks for sharing! Seems to work well from the song I uploaded. Would be nice to have some example songs preloaded on the web UI to see what it looks like without having to upload.
What was the hardest part of this stack? It sounds like porting the weights to Rust via burn was straightforward. Curious if WASM gave you any difficulties.
Checkout the README on GitHub! It has an example split.
Honestly the WASM part was pretty straightforward, other than some shader performance issues with Burn + CubeCL , but nothing this project can really change.
The hardest part was the weight porting honestly, since I had to perfectly recreate all of the digital signal processing and model, any deviations from the original PyTorch implementation, and you ended up with subtle loss of signal in the outputs. Added that folder `bench/` to step through the two implementations piece by piece to help debug where they were diverging -- and it's still not a perfect replication of the original after all the tuning!