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

Files

Steven Lovegrove ed48f54b54 Expand split ops (#2505 )

* candle-onnx: Add Split and Expand operators, Fix Where Op

Implemented based on https://github.com/onnx/onnx/blob/main/docs/Operators.md
Test cases based on those examples.

TODO: Should add the remaining Split examples as tests
TODO: Add.test case that motivates Where fix

* candle-onnx: Add ReduceSum operator

Implemented based on https://github.com/onnx/onnx/blob/main/docs/Operators.md
Test cases based on those examples.

TODO: Should add the remaining ReduceSum examples as tests

* candle-onnx: Add ReduceL2 operator

Implemented based on https://github.com/onnx/onnx/blob/main/docs/Operators.md
Test cases based on those examples.

TODO: Should add the remaining ReduceSum examples as tests

* candle-onnx: Fix Clip operator empty string as default arg issue

Optional input args may be signified by an empty string. The length of the input array is not enough because non optional args may follow optional ones.

I encountered this when trying to use the ONNX model found at https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2 for example.

The LSTM op has a utility which I factored to be more generally accessible, and I have used it in the ops I have recently created or debugged.

I believe it is likely that this issue may also manifest in other ops, but I didn't want to change anything that I'm not testing.

* fix formatting

* fix small mistake made during refactor

2024-09-26 22:57:55 +02:00

src

Expand split ops (#2505 )

2024-09-26 22:57:55 +02:00

tests

Expand split ops (#2505 )

2024-09-26 22:57:55 +02:00

build.rs

Add some preliminary ONNX support (#1260 )

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

Cargo.toml

Move the candle version to 0.7.1. (#2495 )

2024-09-22 20:44:39 +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.