5 releases

0.2.4 Oct 2, 2024
0.2.3 Sep 25, 2024
0.2.1 Sep 18, 2024
0.2.0 Sep 18, 2024
0.1.0 Sep 13, 2024

#902 in Machine learning


Used in 2 crates

MIT/Apache

175KB
4.5K SLoC

moshi - rust

Latest version Documentation License

See the top-level README.md for more information.

This provides the Rust backend (both Mimi and Moshi) and client implementation. The Mimi implementation is available through Python bindings, through the rustymimi package.

Requirements

You will need a recent version of the Rust toolchain. To compile GPU support, you will also need the CUDA properly installed for your GPU, in particular with nvcc.

Rust based Mimi with Python bindings

First, a standalone rust based implementation of Mimi is provided, along with Python bindings. This is the one used by moshi_mlx. It is automatically installed with moshi_mlx, but you can install it separately as

# Install from pip:
pip install rustymimi
# Alternatively, if you want to compile the package run from the root of the repo.
maturin dev -r -m rust/mimi-pyo3/Cargo.toml

Rust server

If you don't have ssl certificates yet, generate a key.pem and cert.pem file using the following command.

openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem -days 365 -nodes -subj "/CN=localhost"

In order to run the rust inference server, use the following command from within the this directory:

cargo run --features cuda --bin moshi-backend -r -- --config moshi-backend/config.json standalone

When using macOS, you can replace --features cuda with --features metal.

Alternatively you can use config-q8.json rather than config.json to use the quantified q8 model. You can select a different pretrained model, e.g. Moshika, by changing the "hf_repo" key in either file.

Once the server has printed 'standalone worker listening', you can use the web UI. By default the rust version uses https so it will be at localhost:8998.

You will get some warnings about the site being unsafe. When using chrome you can bypass it by selecting "Details" or "Advanced", then "Visit this unsafe site" or "Proceed to localhost (unsafe)".

Rust client

We recommend using the web UI as it provides some echo cancellation that helps the overall model quality. Alternatively we provide some command line interfaces for the rust and python versions, the protocol is the same as with the web UI so there is nothing to change on the server side.

Rust Command Line

From within the rust directory, run the following:

cargo run --bin moshi-cli -r -- tui --host localhost

License

The present code is provided under the Apache license.

Dependencies

~13–24MB
~409K SLoC