Fix computing features for whisper. (#38)

csukuangfj · Feb 29, 2024 · cd20baa · cd20baa
1 parent 8142b24
commit cd20baa
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -25,7 +25,7 @@ cmake_minimum_required(VERSION 3.3 FATAL_ERROR)
 
 project(kaldi-native-fbank CXX C)
 
-set(KALDI_NATIVE_FBANK_VERSION "1.18.6")
+set(KALDI_NATIVE_FBANK_VERSION "1.18.7")
 
 # Disable warning about
 #

diff --git a/kaldi-native-fbank/csrc/mel-computations.cc b/kaldi-native-fbank/csrc/mel-computations.cc
@@ -237,6 +237,7 @@ void MelBanks::InitKaldiMelBanks(const MelBanksOptions &opts,
 void MelBanks::InitLibrosaMelBanks(const MelBanksOptions &opts,
                                    const FrameExtractionOptions &frame_opts,
                                    float vtln_warp_factor) {
+  htk_mode_ = opts.htk_mode;
   int32_t num_bins = opts.num_bins;
   if (num_bins < 3) {
     KNF_LOG(FATAL) << "Must have at least 3 mel bins";

diff --git a/kaldi-native-fbank/csrc/mel-computations.h b/kaldi-native-fbank/csrc/mel-computations.h
@@ -165,8 +165,8 @@ class MelBanks {
   std::vector<std::pair<int32_t, std::vector<float>>> bins_;
 
   // TODO(fangjun): Remove debug_ and htk_mode_
-  bool debug_;
-  bool htk_mode_;
+  bool debug_ = false;
+  bool htk_mode_ = false;
 };
 
 }  // namespace knf