feat: Added ndarray-safetensors

This commit is contained in:
uttarayan21
2025-08-19 18:33:38 +05:30
parent 61466c9edd
commit 47218fa696
8 changed files with 1285 additions and 7 deletions

48
Cargo.lock generated
View File

@@ -269,6 +269,20 @@ name = "bytemuck"
version = "1.23.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3995eaeebcdf32f91f980d360f78732ddc061097ab4e39991ae7a6ace9194677"
dependencies = [
"bytemuck_derive",
]
[[package]]
name = "bytemuck_derive"
version = "1.10.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4f154e572231cb6ba2bd1176980827e3d5dc04cc183a75dea38109fbdd672d29"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]]
name = "byteorder-lite"
@@ -504,7 +518,9 @@ dependencies = [
"nalgebra",
"ndarray",
"ndarray-image",
"ndarray-math",
"ndarray-resize",
"ndarray-safetensors",
"ordered-float",
"ort",
"rusqlite",
@@ -830,6 +846,7 @@ version = "2.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "459196ed295495a68f7d7fe1d84f6c4b7ff0e21fe3017b2f283c6fac3ad803c9"
dependencies = [
"bytemuck",
"cfg-if",
"crunchy",
]
@@ -1414,6 +1431,16 @@ dependencies = [
"ndarray",
]
[[package]]
name = "ndarray-math"
version = "0.1.0"
source = "git+https://git.darksailor.dev/servius/ndarray-math#f047966f20835267f20e5839272b9ab36c445796"
dependencies = [
"ndarray",
"num",
"thiserror 2.0.15",
]
[[package]]
name = "ndarray-resize"
version = "0.1.0"
@@ -1426,6 +1453,17 @@ dependencies = [
"thiserror 2.0.15",
]
[[package]]
name = "ndarray-safetensors"
version = "0.1.0"
dependencies = [
"bytemuck",
"half",
"ndarray",
"safetensors",
"thiserror 2.0.15",
]
[[package]]
name = "new_debug_unreachable"
version = "1.0.6"
@@ -1983,6 +2021,16 @@ dependencies = [
"bytemuck",
]
[[package]]
name = "safetensors"
version = "0.6.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "172dd94c5a87b5c79f945c863da53b2ebc7ccef4eca24ac63cca66a41aab2178"
dependencies = [
"serde",
"serde_json",
]
[[package]]
name = "scopeguard"
version = "1.2.0"