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

Files

B1rtek cb180eb23a ONNX: add ArgMin, ArgMax and LeakyRelu (#2246 )

* Add basic RandomUniform implementation

* Use is_some to check if seed is present

* Added Exp operator implementation

* Added ArgMin operator implementation

* Added tests for ArgMin

* ArgMin now returns a tensor with i64

* Added tests from pytorch examples

* Added ArgMax operator implementation

* Added tests for ArgMax

* Added LeakyRelu implementation

* Added a test for LeakyRelu

* Typo fix

* Fix a weird automatic RustRover change

---------

Co-authored-by: Mateusz Okulus <mmokulus@gmail.com>

2024-06-04 22:49:02 +02:00

src

ONNX: add ArgMin, ArgMax and LeakyRelu (#2246 )

2024-06-04 22:49:02 +02:00

tests

ONNX: add ArgMin, ArgMax and LeakyRelu (#2246 )

2024-06-04 22:49:02 +02:00

build.rs

Add some preliminary ONNX support (#1260 )

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

Cargo.toml

Bump the version number to 0.5.1. (#2155 )

2024-05-03 11:17:05 +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.