flystem
/
flystem-usls
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix CPU ep

This commit is contained in:
Jamjamjon 2024-07-17 14:25:28 +08:00 committed by GitHub
parent 56aed9e50c
commit 7740761345
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/engine.rs
View File

@ -246,7 +246,7 @@ impl OrtEngine {
} }
fn build_cpu(builder: &SessionBuilder) -> Result<()> { 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() { if ep.is_available()? && ep.register(builder).is_ok() {
Ok(()) Ok(())
} else { } else {