diff --git a/sbv2_core/Cargo.toml b/sbv2_core/Cargo.toml index 0a8a88c..63badb6 100644 --- a/sbv2_core/Cargo.toml +++ b/sbv2_core/Cargo.toml @@ -31,3 +31,4 @@ cuda_tf32 = [] dynamic = ["ort/load-dynamic"] directml = ["ort/directml"] tensorrt = ["ort/tensorrt"] +coreml = ["ort/coreml"] \ No newline at end of file diff --git a/sbv2_core/src/model.rs b/sbv2_core/src/model.rs index 2fa75c3..2b975ca 100644 --- a/sbv2_core/src/model.rs +++ b/sbv2_core/src/model.rs @@ -35,6 +35,10 @@ pub fn load_model>(model_file: P, bert: bool) -> Result { exp.push(ort::DirectMLExecutionProvider::default().build()); } + #[cfg(feature = "coreml")] + { + exp.push(ort::CoreMLExecutionProvider::default().build()); + } exp.push(ort::CPUExecutionProvider::default().build()); Ok(Session::builder()? .with_execution_providers(exp)?