From 36319dec5d75a7dfe3e3de37b9ca2a76cd52b7b2 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Thu, 19 Dec 2024 17:35:15 +0200 Subject: [PATCH] tts : small QoL for easy model fetch (#10903) --- common/arg.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/common/arg.cpp b/common/arg.cpp index e5ddd8318f787..c3d66efb5a701 100644 --- a/common/arg.cpp +++ b/common/arg.cpp @@ -2206,5 +2206,17 @@ common_params_context common_params_parser_init(common_params & params, llama_ex } ).set_examples({LLAMA_EXAMPLE_TTS, LLAMA_EXAMPLE_SERVER})); + // model-specific + add_opt(common_arg( + {"--tts-oute-default"}, + string_format("use default OuteTTS models (note: can download weights from the internet)"), + [](common_params & params) { + params.hf_repo = "OuteAI/OuteTTS-0.2-500M-GGUF"; + params.hf_file = "OuteTTS-0.2-500M-Q8_0.gguf"; + params.vocoder.hf_repo = "ggml-org/WavTokenizer"; + params.vocoder.hf_file = "WavTokenizer-Large-75-F16.gguf"; + } + ).set_examples({LLAMA_EXAMPLE_TTS})); + return ctx_arg; }