Skip to content

Instantly share code, notes, and snippets.

@a-agmon
Created December 21, 2023 09:14
Show Gist options
  • Save a-agmon/2b510c1d7d4e6f9a14cbcf2e052cccb0 to your computer and use it in GitHub Desktop.
Save a-agmon/2b510c1d7d4e6f9a14cbcf2e052cccb0 to your computer and use it in GitHub Desktop.
loading embeddings
let embeddings = match embeddings_filename.is_empty() {
true => {
println!("no file name provided");
Tensor::new(&[0.0], &device)?
}
false => {
let tensor_file = safetensors::load(embeddings_filename, &device)?;
tensor_file
.get(embeddings_key)
.expect("error getting key:embedding")
.clone()
}
};
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment