search

bes-dev / stable_diffusion.openvino

Apache License 2.0
1.53k stars 208 forks source link

Hi, can u share the code how to export origina weights to onnx? #26

Closed luohao123 closed 1 year ago

luohao123 commented 1 year ago

Hi, can u share the code how to export origina weights to onnx?

bes-dev commented 1 year ago

@luohao123 currently it is not ready ti public release. We will release onnx exporter soon.

luohao123 commented 1 year ago

I have made one: https://github.com/luohao123/gaintmodels/blob/master/export_df_onnx.py

luohao123 commented 1 year ago

@bes-dev Hi, I found my exported onnnx inference result is wrong, can u release your version..

luohao123 commented 1 year ago

ALSO i meet a lot of problemzs when exporting the onnx. For excample the triu op can not erxport and the model of unet is very huge. can not export.

bes-dev commented 1 year ago

@luohao123 currently diffusers library should support onnx export by default.