This commit is contained in:
hlhr202 2024-05-16 14:10:03 +08:00
parent cd6a633fc5
commit bf5a08dd1d
2 changed files with 4 additions and 5 deletions

View file

@ -81,7 +81,7 @@ pub unsafe fn set_log_callback(
log_callback: crate::WhisperLogCallback, log_callback: crate::WhisperLogCallback,
user_data: *mut std::ffi::c_void, user_data: *mut std::ffi::c_void,
) { ) {
unsafe { unsafe {
whisper_rs_sys::whisper_log_set(log_callback, user_data); whisper_rs_sys::whisper_log_set(log_callback, user_data);
#[cfg(feature = "metal")] #[cfg(feature = "metal")]
{ {

View file

@ -76,14 +76,13 @@ fn main() {
let _: u64 = std::fs::copy("src/bindings.rs", out.join("bindings.rs")) let _: u64 = std::fs::copy("src/bindings.rs", out.join("bindings.rs"))
.expect("Failed to copy bindings.rs"); .expect("Failed to copy bindings.rs");
} else { } else {
let bindings = bindgen::Builder::default() let bindings = bindgen::Builder::default().header("wrapper.h");
.header("wrapper.h");
#[cfg(feature = "metal")] #[cfg(feature = "metal")]
let bindings = bindings.header("whisper.cpp/ggml-metal.h"); let bindings = bindings.header("whisper.cpp/ggml-metal.h");
let bindings = bindings.clang_arg("-I./whisper.cpp") let bindings = bindings
.clang_arg("-I./whisper.cpp")
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new())) .parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
.generate(); .generate();