whisper : return with error from whisper_encode_internal and whisper_…

…decode_internal when abort callback is true (ggerganov#1456) Co-authored-by: Ben Nortier <ben@bjnortier.com>
felrock · Nov 18, 2023 · f245740 · f245740
1 parent e3c1340
commit f245740
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/whisper.cpp b/whisper.cpp
@@ -2027,7 +2027,7 @@ static bool whisper_encode_internal(
     wstate.t_encode_us += ggml_time_us() - t_start_us;
     wstate.n_encode++;
 
-    return true;
+    return !(abort_callback && abort_callback(abort_callback_data));
 }
 
 static struct ggml_cgraph * whisper_build_graph_decoder(
@@ -2447,7 +2447,7 @@ static bool whisper_decode_internal(
         wstate.n_prompt++;
     }
 
-    return true;
+    return !(abort_callback && abort_callback(abort_callback_data));
 }