From 908e384cfa0523d761fdb63abd30ca721c4b9fb5 Mon Sep 17 00:00:00 2001 From: Googlefan Date: Sat, 22 Feb 2025 09:29:15 +0000 Subject: [PATCH] fix --- crates/sbv2_core/src/model.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")] {