save embed

cn9.rs

let results: Vec<Result<Tensor, _>> = sentences
    .par_chunks(350)
    .map(|chunk| bert_model.create_embeddings(chunk.to_vec()))
    .collect();

let embeddings = Tensor::cat(
    &results
        .iter()
        .map(|r| r.as_ref().unwrap())
        .collect::<Vec<_>>(),
    0,).unwrap();

embeddings
    .save_safetensors("my_embedding", "embeddings.bin")
    .unwrap();

println!("embeddings.bin saved");

}