Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 11 additions & 1 deletion src/whisper.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -6800,6 +6800,16 @@ int whisper_full_with_state(

result_all.clear();

// if not explicitly set by user, scale audio_ctx for short recordings
if (params.audio_ctx == 0) {
const double audio_length_seconds = (double)n_samples / WHISPER_SAMPLE_RATE;
if (audio_length_seconds <= 30.0) {
int audio_context = std::max(768, std::min((int)(audio_length_seconds / 30.0 * 1500.0) + 128, 1500));
params.audio_ctx = audio_context;
WHISPER_LOG_DEBUG("%s: Audio context length: %d for %.1fs of audio\n", __func__, audio_context, audio_length_seconds);
}
}

if (n_samples > 0) {
// compute log mel spectrogram
if (whisper_pcm_to_mel_with_state(ctx, state, samples, n_samples, params.n_threads) != 0) {
Expand Down Expand Up @@ -9000,4 +9010,4 @@ static void whisper_log_callback_default(ggml_log_level level, const char * text
#endif
fputs(text, stderr);
fflush(stderr);
}
}