diff --git a/crates/sbv2_core/src/model.rs b/crates/sbv2_core/src/model.rs index c1f1e14..f695b23 100644 --- a/crates/sbv2_core/src/model.rs +++ b/crates/sbv2_core/src/model.rs @@ -23,7 +23,7 @@ pub fn load_model>(model_file: P, bert: bool) -> Result #[allow(unused_mut)] let mut cuda = ort::execution_providers::CUDAExecutionProvider::default() .with_conv_algorithm_search( - ort::execution_providers::CUDAExecutionProviderCuDNNConvAlgoSearch::Default, + ort::execution_providers::cuda::CUDAExecutionProviderCuDNNConvAlgoSearch::Default, ); #[cfg(feature = "cuda_tf32")] {