axodox / unpaint

A simple Windows / Xbox app for generating AI images with Stable Diffusion.
Other
268 stars 12 forks source link

Import onnx model #59

Open Cop46 opened 1 month ago

Cop46 commented 1 month ago

Hi, I have used optimum from huggingface to convert the model to onnx and this is impossible to open the model in unpaint because it doesn't contain safety_checker folder.

I have tried to change the code to don't use safety checker but I have one error "Saving initialized tensors (failed:open file model.onnx_data)".... So, impossible to test your software actualy. If you have a valid solution ? Thanks.

axodox commented 1 month ago

You may copy a safety checker onnx from any other model. Eg.: https://huggingface.co/axodoxian/protovision_xl_onnx/tree/main/safety_checker

Cop46 commented 1 month ago

Hi, yes you are right but I think the onnx format or other have changed. Have you tried onnx converted files by "huggingface Optimum" ? Maybe there is a version problem ? Thanks for your good job and sharing your work !