From: arizhih <redacted>
Date: Mon, 15 Jul 2024 12:50:36 +0000 (+0200)
Subject: whisper : fix DTW assert (#2299)
X-Git-Tag: upstream/1.7.4~574
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=7ae885c1efd29b9857cb9aead5561e572c2424b3;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

whisper : fix DTW assert (#2299)
---

diff --git a/src/whisper.cpp b/src/whisper.cpp
index 36370604..47c58cfb 100644
--- a/src/whisper.cpp
+++ b/src/whisper.cpp
@@ -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) {
+    if (ith != 0) {
+        return;
+    }
     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 % 2);
     WHISPER_ASSERT(ggml_n_dims(a) == 3);