From 847f94fdeb9dc1f750ba9deebc763d4d6aedd129 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Sat, 5 Oct 2024 15:22:17 +0300 Subject: [PATCH] whisper : zero-out the KV cache upon clear (#2445) --- src/whisper.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/whisper.cpp b/src/whisper.cpp index 4931b0d..31c6861 100644 --- a/src/whisper.cpp +++ b/src/whisper.cpp @@ -1046,6 +1046,8 @@ static void whisper_kv_cache_clear(struct whisper_kv_cache & cache) { cache.cells[i].seq_id.clear(); } cache.head = 0; + + ggml_backend_buffer_clear(cache.buffer, 0); } static void whisper_kv_cache_seq_rm(