whisper : zero-out the KV cache upon clear (#2445)

ggerganov · Oct 5, 2024 · 47de411 · 47de411
1 parent 0c0247e
commit 47de411
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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(