Compare commits

...

2 Commits

Author SHA1 Message Date
dante
b1481dab69 Update model.rs 2024-10-29 12:01:29 -04:00
dante
693c894a5a chore: bump tract 2024-10-29 11:59:05 -04:00
3 changed files with 30 additions and 18 deletions

44
Cargo.lock generated
View File

@@ -3322,14 +3322,16 @@ dependencies = [
[[package]]
name = "ndarray"
version = "0.15.6"
version = "0.16.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "adb12d4e967ec485a5f71c6311fe28158e9d6f4bc4a447b474184d0f91a8fa32"
checksum = "882ed72dce9365842bf196bdeedf5055305f11fc8c03dee7bb0194a6cad34841"
dependencies = [
"matrixmultiply",
"num-complex",
"num-integer",
"num-traits",
"portable-atomic",
"portable-atomic-util",
"rawpointer",
]
@@ -3857,6 +3859,15 @@ version = "1.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7170ef9988bc169ba16dd36a7fa041e5c4cbeb6a35b76d4c03daded371eae7c0"
[[package]]
name = "portable-atomic-util"
version = "0.2.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "90a7d5beecc52a491b54d6dd05c7a45ba1801666a5baad9fdbfc6fef8d2d206c"
dependencies = [
"portable-atomic",
]
[[package]]
name = "poseidon"
version = "0.2.0"
@@ -5672,8 +5683,8 @@ dependencies = [
[[package]]
name = "tract-core"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"anyhow",
"bit-set",
@@ -5696,8 +5707,8 @@ dependencies = [
[[package]]
name = "tract-data"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"anyhow",
"downcast-rs",
@@ -5711,6 +5722,7 @@ dependencies = [
"nom",
"num-integer",
"num-traits",
"parking_lot",
"scan_fmt",
"smallvec",
"string-interner",
@@ -5718,8 +5730,8 @@ dependencies = [
[[package]]
name = "tract-hir"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"derive-new",
"log",
@@ -5728,8 +5740,8 @@ dependencies = [
[[package]]
name = "tract-linalg"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"byteorder",
"cc",
@@ -5755,8 +5767,8 @@ dependencies = [
[[package]]
name = "tract-nnef"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"byteorder",
"flate2",
@@ -5769,8 +5781,8 @@ dependencies = [
[[package]]
name = "tract-onnx"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"bytes",
"derive-new",
@@ -5786,8 +5798,8 @@ dependencies = [
[[package]]
name = "tract-onnx-opl"
version = "0.21.6-pre"
source = "git+https://github.com/sonos/tract/?rev=40c64319291184814d9fea5fdf4fa16f5a4f7116#40c64319291184814d9fea5fdf4fa16f5a4f7116"
version = "0.21.8-pre"
source = "git+https://github.com/sonos/tract/?rev=9b68101a5d13a73cdd1566e40404900a6367cc47#9b68101a5d13a73cdd1566e40404900a6367cc47"
dependencies = [
"getrandom",
"log",

View File

@@ -74,7 +74,7 @@ tokio = { version = "1.35.0", default-features = false, features = ["macros", "r
pyo3 = { version = "0.21.2", features = ["extension-module", "abi3-py37", "macros"], default-features = false, optional = true }
pyo3-asyncio = { git = "https://github.com/jopemachine/pyo3-asyncio/", branch="migration-pyo3-0.21", features = ["attributes", "tokio-runtime"], default-features = false, optional = true }
pyo3-log = { version = "0.10.0", default-features = false, optional = true }
tract-onnx = { git = "https://github.com/sonos/tract/", rev = "40c64319291184814d9fea5fdf4fa16f5a4f7116", default-features = false, optional = true }
tract-onnx = { git = "https://github.com/sonos/tract/", rev = "9b68101a5d13a73cdd1566e40404900a6367cc47", default-features = false, optional = true }
tabled = { version = "0.12.0", optional = true }
metal = { git = "https://github.com/gfx-rs/metal-rs", optional = true }
objc = { version = "0.2.4", optional = true }

View File

@@ -656,7 +656,7 @@ impl Model {
let mut symbol_values = SymbolValues::default();
for (symbol, value) in run_args.variables.iter() {
let symbol = model.symbol_table.sym(symbol);
let symbol = model.symbols.sym(symbol);
symbol_values = symbol_values.with(&symbol, *value as i64);
debug!("set {} to {}", symbol, value);
}