diff --git a/encodec.cpp b/encodec.cpp index 8c70330..979db26 100644 --- a/encodec.cpp +++ b/encodec.cpp @@ -790,9 +790,9 @@ static struct ggml_cgraph * encodec_build_graph( return gf; } -static bool encodec_model_eval( - std::vector & raw_audio, +bool encodec_model_eval( encodec_context & ectx, + std::vector & raw_audio, int n_threads) { const int64_t t_start_ms = ggml_time_ms(); diff --git a/encodec.h b/encodec.h index 54c1966..1b6df08 100644 --- a/encodec.h +++ b/encodec.h @@ -166,4 +166,9 @@ struct encodec_model encodec_load_model_from_file(std::string fname); struct encodec_context encodec_new_context_with_model(encodec_model & model); +bool encodec_model_eval( + encodec_context & ectx, + std::vector & raw_audio, + int n_threads); + void encodec_free(encodec_context & ectx); \ No newline at end of file