From 77407613456e33864d4116f7b9bab6489c0ce69d Mon Sep 17 00:00:00 2001
From: Jamjamjon <51357717+jamjamjon@users.noreply.github.com>
Date: Wed, 17 Jul 2024 14:25:28 +0800
Subject: [PATCH] Fix CPU ep

---
 src/core/engine.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/core/engine.rs b/src/core/engine.rs
index c0b18aa..59181b3 100644
--- 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 {