From f834ad861dbaa896eb636ab3bca00b8034212a4e Mon Sep 17 00:00:00 2001
From: Dadoou <33223302+Dadoou@users.noreply.github.com>
Date: Tue, 19 Dec 2023 15:55:20 +0800
Subject: [PATCH] Update whisper-fbank.h: correct Dim() function (#88)

---
 kaldifeat/csrc/whisper-fbank.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kaldifeat/csrc/whisper-fbank.h b/kaldifeat/csrc/whisper-fbank.h
index 6985bd5..6d0f640 100644
--- a/kaldifeat/csrc/whisper-fbank.h
+++ b/kaldifeat/csrc/whisper-fbank.h
@@ -51,7 +51,7 @@ class WhisperFbankComputer {
   // are ignored
   explicit WhisperFbankComputer(const WhisperFbankOptions &opts = {});
 
-  int32_t Dim() const { return 80; }
+  int32_t Dim() const { return opts_.num_mels; }
 
   const FrameExtractionOptions &GetFrameOptions() const {
     return opts_.frame_opts;