Update whisper-fbank.h: correct Dim() function (#88)

This commit is contained in:
Dadoou 2023-12-19 15:55:20 +08:00 committed by GitHub
parent 352aa43ce3
commit f834ad861d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -51,7 +51,7 @@ class WhisperFbankComputer {
// are ignored // are ignored
explicit WhisperFbankComputer(const WhisperFbankOptions &opts = {}); explicit WhisperFbankComputer(const WhisperFbankOptions &opts = {});
int32_t Dim() const { return 80; } int32_t Dim() const { return opts_.num_mels; }
const FrameExtractionOptions &GetFrameOptions() const { const FrameExtractionOptions &GetFrameOptions() const {
return opts_.frame_opts; return opts_.frame_opts;