search

kamalkraj / stable-diffusion-tritonserver

Deploy stable diffusion model with onnx/tenorrt + tritonserver
Apache License 2.0
122 stars 19 forks source link

onnx conversion #3

Closed harishprabhala closed 1 year ago

harishprabhala commented 1 year ago

Hi Kamal - I was wondering how you have converted the diffusion model to onnx. I have tried the official implementation from here but it doesn't seem to produce the right onnx file.

https://github.com/huggingface/diffusers/blob/main/scripts/convert_stable_diffusion_checkpoint_to_onnx.py

Can you please let me know how you have done the conversion.

kamalkraj commented 1 year ago

Try v2 branch - WIP

Please let me know if any issues