remove cuda error

StarlightSearch · Dec 7, 2024 · edaa279 · edaa279
1 parent 8694bcb
commit edaa279
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rust/src/embeddings/local/bert.rs b/rust/src/embeddings/local/bert.rs
@@ -114,7 +114,7 @@ impl OrtBertEmbedder {
         let threads = std::thread::available_parallelism().unwrap().get();
         let model = Session::builder()?
             .with_execution_providers([
-                CUDAExecutionProvider::default().build().error_on_failure(),
+                CUDAExecutionProvider::default().build(),
                 CoreMLExecutionProvider::default().build(),
             ])?
             .with_optimization_level(GraphOptimizationLevel::Level3)?

diff --git a/rust/src/embeddings/local/jina.rs b/rust/src/embeddings/local/jina.rs
@@ -115,7 +115,7 @@ impl OrtJinaEmbedder {
         let threads = std::thread::available_parallelism().unwrap().get();
         let session = Session::builder()?
             .with_execution_providers([
-                CUDAExecutionProvider::default().build().error_on_failure(),
+                CUDAExecutionProvider::default().build(),
                 CoreMLExecutionProvider::default().build(),
             ])?
             .with_optimization_level(GraphOptimizationLevel::Level3)?