Fix CPU ep

jamjamjon · Jul 17, 2024 · 7740761 · 7740761
1 parent 56aed9e
commit 7740761
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/core/engine.rs b/src/core/engine.rs
@@ -246,7 +246,7 @@ impl OrtEngine {
     }
 
     fn build_cpu(builder: &SessionBuilder) -> Result<()> {
-        let ep = ort::CUDAExecutionProvider::default();
+        let ep = ort::CPUExecutionProvider::default();
         if ep.is_available()? && ep.register(builder).is_ok() {
             Ok(())
         } else {