rust/sbv2-api
1
0
Fork 0
mirror of https://github.com/neodyland/sbv2-api.git synced 2026-01-06 14:32:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: level

Browse Source
This commit is contained in:
tuna2134
2024-09-09 09:55:50 +00:00
parent e5d6ec1eed
commit f4f2aef534
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sbv2_core/src/bert.rs
View File

@@ -4,7 +4,7 @@ use ort::{GraphOptimizationLevel, Session};
pub fn load_model() -> Result<Session> { pub fn load_model() -> Result<Session> {
let session = Session::builder()? let session = Session::builder()?
.with_optimization_level(GraphOptimizationLevel::Level1)? .with_optimization_level(GraphOptimizationLevel::Level3)?
.with_intra_threads(1)? .with_intra_threads(1)?
.commit_from_file("models/debert.onnx")?; .commit_from_file("models/debert.onnx")?;
Ok(session) Ok(session)