mirror of
https://github.com/PABannier/bark.cpp
synced 2026-03-18 04:51:41 +01:00
65 lines
2.0 KiB
C++
65 lines
2.0 KiB
C++
#include <iostream>
|
|
#include <tuple>
|
|
|
|
#include "bark.h"
|
|
#include "common.h"
|
|
#include "ggml.h"
|
|
|
|
int main(int argc, char **argv) {
|
|
ggml_time_init();
|
|
const int64_t t_main_start_us = ggml_time_us();
|
|
|
|
bark_params params;
|
|
bark_verbosity_level verbosity = bark_verbosity_level::LOW;
|
|
|
|
if (bark_params_parse(argc, argv, params) > 0) {
|
|
fprintf(stderr, "%s: Could not parse arguments\n", __func__);
|
|
return 1;
|
|
}
|
|
|
|
std::cout << R"( __ __ )"
|
|
<< "\n"
|
|
<< R"( / /_ ____ ______/ /__ _________ ____ )"
|
|
<< "\n"
|
|
<< R"( / __ \/ __ `/ ___/ //_/ / ___/ __ \/ __ \)"
|
|
<< "\n"
|
|
<< R"( / /_/ / /_/ / / / ,< _ / /__/ /_/ / /_/ /)"
|
|
<< "\n"
|
|
<< R"(/_.___/\__,_/_/ /_/|_| (_) \___/ .___/ .___/ )"
|
|
<< "\n"
|
|
<< R"( /_/ /_/ )"
|
|
<< "\n";
|
|
|
|
// initialize bark context
|
|
struct bark_context *bctx = bark_load_model(params.model_path, verbosity);
|
|
if (!bctx) {
|
|
fprintf(stderr, "%s: Could not load model\n", __func__);
|
|
exit(1);
|
|
}
|
|
|
|
bctx->encodec_model_path = params.encodec_model_path;
|
|
|
|
// generate audio
|
|
if (!bark_generate_audio(bctx, params.prompt, params.dest_wav_path, params.n_threads)) {
|
|
fprintf(stderr, "%s: An error occured. If the problem persists, feel free to open an issue to report it.\n", __func__);
|
|
exit(1);
|
|
}
|
|
|
|
auto &audio_arr = bctx->audio_arr;
|
|
write_wav_on_disk(audio_arr, params.dest_wav_path);
|
|
|
|
// report timing
|
|
{
|
|
const int64_t t_main_end_us = ggml_time_us();
|
|
|
|
printf("\n\n");
|
|
printf("%s: load time = %8.2f ms\n", __func__, bctx->t_load_us / 1000.0f);
|
|
printf("%s: eval time = %8.2f ms\n", __func__, bctx->t_eval_us / 1000.0f);
|
|
printf("%s: total time = %8.2f ms\n", __func__, (t_main_end_us - t_main_start_us) / 1000.0f);
|
|
}
|
|
|
|
bark_free(bctx);
|
|
|
|
return 0;
|
|
}
|