diff --git a/sbv2_core/src/bert.rs b/sbv2_core/src/bert.rs index a5c30b2..3377bdc 100644 --- a/sbv2_core/src/bert.rs +++ b/sbv2_core/src/bert.rs @@ -4,7 +4,7 @@ use ort::{GraphOptimizationLevel, Session}; pub fn load_model() -> Result { let session = Session::builder()? - .with_optimization_level(GraphOptimizationLevel::Level1)? + .with_optimization_level(GraphOptimizationLevel::Level3)? .with_intra_threads(1)? .commit_from_file("models/debert.onnx")?; Ok(session)