diff --git a/modules/llama-swap.nix b/modules/llama-swap.nix index 2ef51ae..ec261b8 100644 --- a/modules/llama-swap.nix +++ b/modules/llama-swap.nix @@ -129,18 +129,17 @@ ]; macros.model = "${whisper-models}/models/nb-whisper-small-q5_0.bin"; }; - "fish-speech-s2-pro" = { - cmd = "${pkgs.bash}/bin/bash -c '${podman} pull fishaudio/fish-speech:server-cpu && exec ${podman} run --rm --name fish-speech-$\{PORT\} -p $\{PORT\}:8080 -v $\{models\}:/app/checkpoints:Z fishaudio/fish-speech:server-cpu'"; - cmdStop = "${podman} stop fish-speech-$\{PORT\}"; - checkEndpoint = "/v1/health"; + "kokoro-tts" = { + cmd = "${pkgs.bash}/bin/bash -c '${podman} pull ghcr.io/remsky/kokoro-fastapi-cpu:latest && exec ${podman} run --rm --name kokoro-tts-$\{PORT\} -p $\{PORT\}:8880 ghcr.io/remsky/kokoro-fastapi-cpu:latest'"; + cmdStop = "${podman} stop kokoro-tts-$\{PORT\}"; + checkEndpoint = "/v1/audio/voices"; proxy = "http://127.0.0.1:$\{PORT\}"; ttl = 0; aliases = [ "tts-1" "tts" - "fish-speech" + "kokoro" ]; - macros.models = "${fish-speech-models}/models"; }; }; };