Open Goody27 opened 1 year ago
Thanks for opening! We support instance norm in Axon so this should be straightforward enough to implement, I will take a look this weekend
I've added instance norm on master, but importing the VAE doesn't work, there's some strange things going on that don't seem to respect the ONNX spec. I will look further into it
Thank you for your research.
I'm trying to get StableDiffusion working using AxonOnnx.
Reproduction steps
https://huggingface.co/bes-dev/stable-diffusion-v1-4-onnx/raw/main/vae_encoder.onnx
intomodels/onnx/vae_encoder.onnx
.{model, params} = AxonOnnx.import("models/onnx/vae_encoder.onnx")
on iex or Livebook.Environment