tomesd icon indicating copy to clipboard operation
tomesd copied to clipboard

Request: Add compatiblity with ONNX (OnnxStableDiffusionPipeline)

Open n00mkrad opened this issue 2 years ago • 2 comments

Would be really great to get a speedup on DirectML/ONNX pipelines from Diffusers.

E.g. OnnxStableDiffusionPipeline, OnnxStableDiffusionImg2ImgPipeline, OnnxStableDiffusionInpaintPipeline.

n00mkrad avatar Apr 28 '23 13:04 n00mkrad

I'm not familiar with ONNX. What are the main differences with regular pytorch? If it's easy to do, I can write a patch.

dbolya avatar May 07 '23 21:05 dbolya

Hi @dbolya, @n00mkrad , I've implemented an ONNX friendly counterpart :) Let me know how that works for you

NatanBagrov avatar May 30 '23 13:05 NatanBagrov