feat: Save tensors to database as safetensor

This commit is contained in:
uttarayan21
2025-08-19 18:35:08 +05:30
parent 47218fa696
commit 37adb74adf

View File

@@ -308,9 +308,11 @@ impl FaceDatabase {
embedding: ndarray::ArrayView1<f32>, embedding: ndarray::ArrayView1<f32>,
model_name: &str, model_name: &str,
) -> Result<i64> { ) -> Result<i64> {
// Convert f32 slice to bytes let embedding_bytes =
// let embedding_bytes: Vec<u8> = embedding.iter().flat_map(|&f| f.to_le_bytes()).collect(); ndarray_safetensors::SafeArrays::from_ndarrays([("embedding", embedding)])
let embedding_bytes = ndarray_safetensors::SafeArrays::new(); .change_context(Error)?
.serialize()
.change_context(Error)?;
let mut stmt = self let mut stmt = self
.conn .conn