From 080b77f2b9245358bbd0f00cc007dcf78d81d4f8 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Bannier <pierreantoine.bannier@gmail.com>
Date: Sun, 22 Oct 2023 15:53:47 +0200
Subject: [PATCH] fix number of codebooks

---
 encodec.cpp | 1 +
 encodec.h   | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/encodec.cpp b/encodec.cpp
index 76ba54b..d85d45a 100644
--- a/encodec.cpp
+++ b/encodec.cpp
@@ -1159,6 +1159,7 @@ struct encodec_context * encodec_load_model(const std::string & model_path) {
     }
 
     ectx->model.hparams.n_q = get_num_codebooks(bandwidth, hop_length, sr);
+    fprintf(stderr, "%s: n_q = %d\n", __func__, ectx->model.hparams.n_q);
 
     ectx->t_load_us = ggml_time_us() - t_start_load_us;
 
diff --git a/encodec.h b/encodec.h
index d7e1d93..bcf7bf3 100644
--- a/encodec.h
+++ b/encodec.h
@@ -43,7 +43,7 @@ struct encodec_hparams {
     int32_t bandwidth = 24;
 
     // The number of codebooks.
-    int32_t n_q;
+    int32_t n_q = 32;
 
     int32_t ftype;
 };