vall-e/vall_e.cpp/README.md

46 lines
1.7 KiB
Markdown
Raw Normal View History

# vall_e.cpp
This is an implementation that makes use of [llama.cpp](https://github.com/ggerganov/llama.cpp/) and [encodec.cpp](https://github.com/PABannier/encodec.cpp).
At the moment it's ***very*** barebones as I try and wrestle with `llama.cpp`'s API without needing to modify its code.
## Build
2024-12-21 21:48:12 +00:00
Populate `./include/` with the `llama.cpp` and `encodec.cpp` headers.
2024-12-21 21:48:12 +00:00
Populate `./libs/` with the compiled libraries of `llama.cpp` and `encodec.cpp`.
Run `make`.
2024-12-22 21:05:45 +00:00
### Required Modifications
[`encodec.cpp`](https://github.com/e-c-k-e-r/encodec.cpp) requires updating its GGML copy to the latest version, which requires a few lines to get the CPU backend working.
[`llama.cpp`](https://github.com/e-c-k-e-r/llama.cpp) only possible modification needs to ensure that a non-causal attention mask is used; everything necessary can be hacked together with clever tricks.
2024-12-22 21:05:45 +00:00
## To-Do
* [x] converted model to GGUF
2024-12-22 01:59:56 +00:00
* [ ] convert it without modifying any of the existing code, as the tokenizer requires some care
* [ ] *actually* convert the model properly, as the embeddings differ from the real model
* [x] basic framework
2024-12-21 17:56:22 +00:00
* [x] load the quantized model
* [x] orchestrate the required embeddings
* [x] juggle the output head / classifier properly
* [ ] phonemize text
2024-12-22 01:59:56 +00:00
* with the help of espeak-ng
* [ ] tokenize phonemes
2024-12-22 01:59:56 +00:00
* the tokenizer is being a huge thorn on actual sequences
2024-12-22 01:16:44 +00:00
* [x] load audio from disk
* [x] encode audio
* [x] sum embeddings for the `prom` and prior `resp`s
* [ ] working `AR` output
* [x] `AR` sampling
* [ ] working `NAR-len` output
* [x] `NAR-len` sampling
* [ ] working `NAR` output
* [x] `NAR` sampling
2024-12-22 01:59:56 +00:00
* [x] decode audio to disk
2024-12-22 01:16:44 +00:00
* [ ] a functional CLI
* [ ] actually make it work