feat: Save tensors to database as safetensor
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user