mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00

Files

shua a8288b7a72 onnx: workaround pow with negative base (#2439 )

* onnx: workaround pow with negative base

rather than fully defining pow in the cpu backend (as in #2318),
this implements a much smaller change which is sufficient to evaluate silero-vad
onnx models. Specifically, checking if pow is run with 2.0 exponent, and if so
evaluate as simply `x*x` instead of the cpu backend of `e^(2.0 * ln(x))`.

* PR: use Tensor::powf insead

powf correctly handles a negative base.

2024-08-22 13:34:53 +02:00

src

onnx: workaround pow with negative base (#2439 )

2024-08-22 13:34:53 +02:00

tests

onnx: implement LSTM op (#2268 )

2024-08-19 09:06:17 +02:00

build.rs

Add some preliminary ONNX support (#1260 )

2023-11-04 06:36:05 +01:00

Cargo.toml

Bump the version to 0.6.1. (#2438 )

2024-08-22 09:23:52 +02:00

README.md

Put the onnx example behind a feature flag. (#1276 )

2023-11-06 07:45:07 +01:00

README.md

candle-onnx

This crate adds ONNX support to candle

FAQ

Missing protoc installation when compiling candle-onnx

The candle-onnx dependency prost-build no longer comes bundled with prost binaries. This could cause the following error when attempting to compile candle-onnx:

error: failed to run custom build command for `candle-onnx`
Caused by: // (...)
  Could not find `protoc` installation and this build crate cannot proceed without this knowledge.

To fix this issue install protoc on your system and make it available in your system PATH. See the protoc documentation for more information.