bentoml / BentoDiffusion

BentoDiffusion: A collection of diffusion models served with BentoML
https://bentoml.com
Apache License 2.0
334 stars 25 forks source link

Support raw prompt embeddings input #17

Open cvb941 opened 1 year ago

cvb941 commented 1 year ago

Hi.

Instead of passing a string prompt to the generator, I would like to pass in the raw text embeddings.

The HF Diffusers library allows this using the prompt_embeds (and pooled_prompt_embeds for SDXL) parameters on the StableDiffusionPipeline pipeline objects.

Would it be possible to add support for this to OneDiffusion? Thank you.