mirror/whisper.cpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

whisper : fix DTW assert (#2299)

Browse Source
This commit is contained in:
arizhih 2024-07-15 14:50:36 +02:00 committed by GitHub
parent d207c68822
commit 7ae885c1ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/whisper.cpp
View File

@ -7220,9 +7220,10 @@ struct median_filter_user_data {
}; };
static void median_filter(struct ggml_tensor * dst , const struct ggml_tensor * a, int ith, int nth, void * userdata) { static void median_filter(struct ggml_tensor * dst , const struct ggml_tensor * a, int ith, int nth, void * userdata) {
if (ith != 0) {
return;
}
int filter_width = ((median_filter_user_data *) userdata)->filter_width; int filter_width = ((median_filter_user_data *) userdata)->filter_width;
WHISPER_ASSERT(nth == 1);
WHISPER_ASSERT(ith == 0);
WHISPER_ASSERT(filter_width < a->ne[2]); WHISPER_ASSERT(filter_width < a->ne[2]);
WHISPER_ASSERT(filter_width % 2); WHISPER_ASSERT(filter_width % 2);
WHISPER_ASSERT(ggml_n_dims(a) == 3); WHISPER_ASSERT(ggml_n_dims(a) == 3);