mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-08-22 15:45:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix CPU ep

Browse Source
This commit is contained in:
Jamjamjon
2024-07-17 14:25:28 +08:00
committed by GitHub
parent 56aed9e50c
commit 7740761345
1 changed files with 1 additions and 1 deletions
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<()> {
let ep = ort::CUDAExecutionProvider::default();
let ep = ort::CPUExecutionProvider::default();
if ep.is_available()? && ep.register(builder).is_ok() {
Ok(())
} else {