huggingface / optimum-neuron

Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.
Apache License 2.0
176 stars 51 forks source link

Add optimum-neuron support for diffusers.StableDiffusionControlNetPipeline #575

Closed yahavb closed 5 days ago

yahavb commented 2 months ago

Feature request

Implement ControlNet in Neuron to control stable diffusion image generation with control image condition like in https://huggingface.co/docs/diffusers/en/api/pipelines/controlnet#diffusers.StableDiffusionControlNetPipeline.__call__.example

Motivation

flexible and accurate way to control the image generation process on AWS Inferentia instances

Your contribution

Happy to contribute test scripts and documentation.

asutermo commented 1 month ago

Any update on this?

HuggingFaceDocBuilderDev commented 3 weeks ago

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread. Thank you!

yahavb commented 2 weeks ago

We still need that indeed.