eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor: opaque Error

Browse Source 
Shaves off the `thiserror` dependency and should improve compile times slightly.
Unfortunately this does mean we can't match on `Error` anymore, though I'm not sure if that was ever useful to begin with.
This commit is contained in:
Carson M.
2024-08-31 15:17:06 -05:00
parent 18feafe83b
commit 2d26f1f363
44 changed files with 428 additions and 699 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/sentence-transformers/examples/semantic-similarity.rs
View File

@@ -1,7 +1,7 @@
use std::path::Path;
use ndarray::{s, Array1, Array2, Axis, Ix2};
use ort::{CUDAExecutionProvider, GraphOptimizationLevel, Session};
use ort::{CUDAExecutionProvider, Error, GraphOptimizationLevel, Session};
use tokenizers::Tokenizer;
/// Example usage of a text embedding model like Sentence Transformers' `all-mini-lm-l6` model for semantic textual
@@ -31,7 +31,7 @@ fn main() -> ort::Result<()> {
let inputs = vec!["The weather outside is lovely.", "It's so sunny outside!", "She drove to the stadium."];
// Encode our input strings. `encode_batch` will pad each input to be the same length.
let encodings = tokenizer.encode_batch(inputs.clone(), false)?;
let encodings = tokenizer.encode_batch(inputs.clone(), false).map_err(|e| Error::new(e.to_string()))?;
// Get the padded length of each encoding.
let padded_token_length = encodings[0].len();