diff options
Diffstat (limited to 'candle-pyo3/Cargo.toml')
| -rw-r--r-- | candle-pyo3/Cargo.toml | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-pyo3/Cargo.toml b/candle-pyo3/Cargo.toml index b0452404..f79277f2 100644 --- a/candle-pyo3/Cargo.toml +++ b/candle-pyo3/Cargo.toml @@ -17,6 +17,7 @@ crate-type = ["cdylib"] accelerate-src = { workspace = true, optional = true } candle = { path = "../candle-core", version = "0.3.0", package = "candle-core" } candle-nn = { path = "../candle-nn", version = "0.3.0" } +candle-onnx = {path= "../candle-onnx", version = "0.3.0", optional = true} half = { workspace = true } intel-mkl-src = { workspace = true, optional = true } pyo3 = { version = "0.20.0", features = ["extension-module", "abi3-py38"] } @@ -29,3 +30,5 @@ default = [] accelerate = ["dep:accelerate-src", "candle/accelerate"] cuda = ["candle/cuda"] mkl = ["dep:intel-mkl-src","candle/mkl"] +onnx = ["dep:candle-onnx"] + |