more verbose errors

PABannier · Jan 7, 2024 · b3d9179 · b3d9179
1 parent 1fbe29d
commit b3d9179
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/bark/bark.cpp b/bark/bark.cpp
@@ -2102,11 +2102,8 @@ bool bark_generate_audio(
  }
  }
 
- fprintf(stderr, "fine_tokens shape: [%zu, %zu]\n", bctx->fine_tokens.size(), bctx->fine_tokens[0].size());
- fprintf(stderr, "encodec_tokens shape: [%zu]\n", encodec_tokens.size());
-
  if (!encodec_decompress_audio(ectx, encodec_tokens, n_threads)) {
- printf("%s: error during audio generation\n", __func__);
+ printf("%s: Could not generate waveform from tokens with Encodec\n", __func__);
  return false;
  }
 

diff --git a/bark/examples/main/main.cpp b/bark/examples/main/main.cpp
@@ -38,7 +38,7 @@ int main(int argc, char **argv) {
 
  // generate audio
  if (!bark_generate_audio(bctx, params.prompt, params.dest_wav_path, params.n_threads, verbosity)) {
- fprintf(stderr, "%s: Could not generate audio\n", __func__);
+ fprintf(stderr, "%s: An error occured. If the problem persists, feel free to open an issue to report it.\n", __func__);
  exit(1);
  }
 

diff --git a/encodec.cpp b/encodec.cpp