vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-07-10 01:18:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

option to save tts as mp3, currently slightly bugged

Browse source
This commit is contained in:
Concedo 2026-06-18 22:16:29 +08:00
parent 2c64520ba6
commit 1b36e7f606
4 changed files with 29 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
otherarch/tts_adapter.cpp
View file

@ -492,6 +492,19 @@ static int code_terminate_id = 151670;
static int nthreads = 4;
static int tts_max_len = 4096;
static std::string save_tts_audio_base64(const std::vector<float> & audio, int sample_rate, bool use_mp3)
{
if (!use_mp3) {
return save_wav16_base64(audio, sample_rate);
}
std::vector<float> stereo_audio;
stereo_audio.reserve(audio.size() * 2);
stereo_audio.insert(stereo_audio.end(), audio.begin(), audio.end());
stereo_audio.insert(stereo_audio.end(), audio.begin(), audio.end());
return save_stereo_mp3_base64(stereo_audio, (int) audio.size(), sample_rate);
}
//ttscpp specific
static bool is_ttscpp_file = false;
static generation_configuration * ttscpp_config = nullptr;
@ -718,7 +731,7 @@ static tts_generation_outputs ttstype_generate_ttscpp(const tts_generation_input
printf("\nTTS Generated audio in %.2fs.\n",ttstime);
std::vector<float> wavdat = std::vector(response_data.data, response_data.data + response_data.n_outputs);
//audio_post_clean(wavdat);
last_generated_audio = save_wav16_base64(wavdat, ttscpp_runner->sampling_rate);
last_generated_audio = save_tts_audio_base64(wavdat, ttscpp_runner->sampling_rate, inputs.use_mp3);
output.data = last_generated_audio.c_str();
output.status = 1;
last_generation_settings_audio_seed = 0;
@ -1131,7 +1144,7 @@ static tts_generation_outputs ttstype_generate_outetts(const tts_generation_inpu
return output;
}
last_generated_audio = save_wav16_base64(audio, t_sr);
last_generated_audio = save_tts_audio_base64(audio, t_sr, inputs.use_mp3);
ttstime = timer_check();
printf("\nTTS Generated %d audio tokens in %.2fs.\n",(int) codes.size(),ttstime);
@ -1239,7 +1252,7 @@ static tts_generation_outputs ttstype_generate_qwen3tts(const tts_generation_inp
ttstime = timer_check();
printf("\nTTS Generated audio in %.2fs.\n",ttstime);
last_generated_audio = save_wav16_base64(result.audio, result.sample_rate);
last_generated_audio = save_tts_audio_base64(result.audio, result.sample_rate, inputs.use_mp3);
output.data = last_generated_audio.c_str();
output.status = 1;
last_generation_settings_audio_seed = inputs.audio_seed;