search

borisdayma / dalle-mini

DALL·E Mini - Generate images from a text prompt
https://www.craiyon.com
Apache License 2.0
14.74k stars 1.21k forks source link

Can we have onnx models for Dalle-mini #227

Open 0xrushi opened 2 years ago

0xrushi commented 2 years ago

I realized this model doesn't use tensorflow/pytorch e.g Pytorch models to onnx https://docs.microsoft.com/en-us/windows/ai/windows-ml/tutorials/pytorch-convert-model

FamousDirector commented 2 years ago

It may be possible to export the jax models to Tensorflow format. See here: https://github.com/google/jax/tree/main/jax/experimental/jax2tf