Skip to content

Commit 40bd16e

Browse files
committed
add system_info in output in section params
1 parent 3b85c42 commit 40bd16e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/rcpp_whisper.cpp

+6-4
Original file line numberDiff line numberDiff line change
@@ -446,10 +446,12 @@ Rcpp::List whisper_encode(SEXP model, std::string path, std::string language,
446446
Rcpp::Named("beam_size") = params.beam_size,
447447
Rcpp::Named("best_of") = params.best_of,
448448
Rcpp::Named("split_on_word") = params.split_on_word,
449-
Rcpp::Named("diarize") = params.diarize));
450-
451-
452-
449+
Rcpp::Named("diarize") = params.diarize,
450+
Rcpp::Named("system_info") = Rcpp::List::create(
451+
Rcpp::Named("n_threads") = params.n_threads,
452+
Rcpp::Named("n_processors") = params.n_processors,
453+
Rcpp::Named("available_concurrency") = std::thread::hardware_concurrency(),
454+
Rcpp::Named("optimisations") = whisper_print_system_info())));
453455
return output;
454456
}
455457

0 commit comments

Comments
 (0)