git/bark.cpp
1
0
Fork 0
mirror of https://github.com/PABannier/bark.cpp synced 2026-03-02 21:21:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

FIX/API Proper constructor for bark_progress (#107)

Browse Source
This commit is contained in:
PAB 2023-09-09 23:50:06 +02:00 committed by GitHub
parent dbe6aacb4f
commit badd010cfa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
bark.cpp
View File

@ -159,13 +159,13 @@ struct bark_context {
struct bark_progress {
float current = 0.0f;
const char * func = NULL;
const char * func;
bark_progress() {}
bark_progress(const char * func): func(func) {}
void callback(float progress) {
float percentage = progress * 100;
if (percentage == 0.0f && func != NULL) {
if (percentage == 0.0f) {
fprintf(stderr, "%s: ", func);
}
while (percentage > current) {
@ -1707,8 +1707,7 @@ void bark_forward_text_encoder(struct bark_context * ctx, int n_threads) {
bark_sequence out;
bark_progress progress;
progress.func = __func__;
bark_progress progress( __func__);
gpt_model * model = &ctx->model.text_model;
@ -1768,8 +1767,7 @@ void bark_forward_coarse_encoder(struct bark_context * ctx, int n_threads) {
bark_codes out_coarse;
bark_sequence out;
bark_progress progress;
progress.func = __func__;
bark_progress progress(__func__);
int max_coarse_history = ctx->max_coarse_history;
int sliding_window_size = ctx->sliding_window_size;
@ -1880,9 +1878,11 @@ void bark_forward_coarse_encoder(struct bark_context * ctx, int n_threads) {
}
void bark_forward_fine_encoder(struct bark_context * ctx, int n_threads) {
// input shape: (N, n_codes)
// input shape: [N, n_codes]
const int64_t t_main_start_us = ggml_time_us();
bark_progress progress(__func__);
bark_codes input = ctx->coarse_tokens;
float temp = ctx->fine_temp;
@ -1892,9 +1892,6 @@ void bark_forward_fine_encoder(struct bark_context * ctx, int n_threads) {
gpt_model * model = &ctx->model.fine_model;
bark_progress progress;
progress.func = __func__;
int n_coarse = input[0].size();
int original_seq_len = input.size();
int n_remove_from_end = 0;