summaryrefslogtreecommitdiff
path: root/candle-pyo3/py_src/candle/onnx
Commit message (Collapse)AuthorAgeFilesLines
* Detach the tensors on batch-norm eval. (#1702)Laurent Mazare2024-02-131-0/+11
| | | | | | | | | | | | | * Detach the tensors on batch-norm eval. * Fix pyo3 bindings. * Black tweak. * Formatting. * Also update the pyo3-onnx formatting. * Apply black.
* PyO3: Add optional `candle.onnx` module (#1282)Lukas Kreussel2023-11-082-0/+94
* Start onnx integration * Merge remote-tracking branch 'upstream/main' into feat/pyo3-onnx * Implement ONNXModel * `fmt` * add `onnx` flag to python ci * Pin `protoc` to `25.0` * Setup `protoc` in wheel builds * Build wheels with `onnx` * Install `protoc` in manylinux containers * `apt` -> `yum` * Download `protoc` via bash script * Back to `manylinux: auto` * Disable `onnx` builds for linux