Skip to content

Commit

Permalink
fix jina
Browse files Browse the repository at this point in the history
  • Loading branch information
akshayballal95 committed Jan 9, 2025
1 parent 3b91bea commit 47c6542
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 4 deletions.
2 changes: 1 addition & 1 deletion Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion rust/Cargo.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[package]
name = "embed_anything"
version = "0.4.17"
version = "0.5.1"
edition.workspace = true
license.workspace = true
description.workspace = true
Expand Down
4 changes: 2 additions & 2 deletions rust/examples/ort_models.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ use std::time::Instant;
async fn main() -> Result<(), anyhow::Error> {
let model = Arc::new(
Embedder::from_pretrained_onnx(
"bert",
Some(ONNXModel::ModernBERTBase),
"jina",
Some(ONNXModel::JINAV3),
None,
None,
Some(Dtype::F16),
Expand Down
3 changes: 3 additions & 0 deletions rust/src/embeddings/local/jina.rs
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,10 @@ impl OrtJinaEmbedder {
None => path.to_string(),
};
let weights = api.get(model_path.as_str());
let _ = api.get(format!("{path}_data").as_str());

(config, tokenizer, weights, tokenizer_config)

};

let weights_filename = match weights_filename {
Expand Down

0 comments on commit 47c6542

Please sign in to comment.