Skip to content

Commit

Permalink
upgrading to beta V 2 of ort with optional compilation of GPU
Browse files Browse the repository at this point in the history
  • Loading branch information
maxwellflitton committed Dec 24, 2024
1 parent 3f29dd5 commit e1eb109
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions modules/core/src/execution/session.rs
Original file line number Diff line number Diff line change
@@ -1,8 +1,16 @@
//! Defines the session module for the execution module.
use ort::session::Session;
use crate::errors::error::{SurrealError, SurrealErrorStatus};
use crate::safe_eject;


/// Creates a session for a model.
///
/// # Arguments
/// * `model_bytes` - The model bytes (usually extracted fromt the surml file)
///
/// # Returns
/// A session object.
pub fn get_session(model_bytes: Vec<u8>) -> Result<Session, SurrealError> {
let builder = safe_eject!(Session::builder(), SurrealErrorStatus::Unknown);

Expand Down

0 comments on commit e1eb109

Please sign in to comment.