axon_onnx
axon_onnx copied to clipboard
elixir-nx
Easily convert models between ONNX and Axon
I am trying to import `microsoft/swin-base-patch4-window12-384` with: ``` AxonOnnx.import("../../models/swin-base-patch4-window12-384-export/model.onnx", batch_size: 32, num_channels: 3, height: 384, width: 384) ``` but I get an error: ``` Bug Bug ..!!** (ArgumentError) length at...
I am getting the following error when attempting to install axon_onnx in livebook. == Compilation error in file lib/axon_onnx/shared.ex == ** (CompileError) lib/axon_onnx/shared.ex:58: undefined function transform/2 (there is no such...
When I pull down the BERT or RoBERTa models from hugging face I can't export them today with 0.4 because the `op` in the Axon.Node doesn't match any of the...
Description: I am facing an issue while attempting to import ONNX files into my project. Whenever I try to import an ONNX file using the provided code snippet, I encounter...
WIP Implementation for [Resize Operator](https://github.com/onnx/onnx/blob/main/docs/Operators.md#resize) fixes https://github.com/elixir-nx/axon_onnx/issues/34
The model [Tiny YOLOv3](https://github.com/onnx/models/tree/main/vision/object_detection_segmentation/tiny-yolov3) raises an ArgumentError when trying to import it: ```elixir ** (ArgumentError) unsupported "Resize" (axon_onnx 0.1.0) lib/axon_onnx/deserialize.ex:1595: AxonOnnx.Deserialize.recur_nodes/2 (elixir 1.13.2) lib/enum.ex:2396: Enum."-reduce/3-lists^foldl/2-0-"/3 (axon_onnx 0.1.0) lib/axon_onnx/deserialize.ex:44: AxonOnnx.Deserialize.graph_to_axon/2 (axon_onnx...
I'm attempting to import a model with a dynamic batch size (modified export of a Whisper encoder) and I'm getting the following error ``` 12:08:45.008 [warning] mel_spectrogram_dynamic_axes_1 has no specified...
Hi! Thanks a lot for this library. I'm new to deep learning and wanted to try out a few models in Elixir. Specifically I'm interested in trying out semantic search...
I'm trying to get StableDiffusion working using AxonOnnx. # Reproduction steps 1. Download the file from `https://huggingface.co/bes-dev/stable-diffusion-v1-4-onnx/raw/main/vae_encoder.onnx` into `models/onnx/vae_encoder.onnx`. 2. Do `{model, params} = AxonOnnx.import("models/onnx/vae_encoder.onnx")` on iex or Livebook. 3....
