Zhendong-Wang / Prompt-Diffusion

Official PyTorch implementation of the paper "In-Context Learning Unlocked for Diffusion Models"
Apache License 2.0
371 stars 9 forks source link

Diffusers support #12

Open iczaw opened 7 months ago

iczaw commented 7 months ago

Would you mind converting your checkpoint to Diffusers format and sharing it on your HF profile?

https://github.com/iczaw/diffusers-pd/blob/prompt-diffusion/examples/research_projects/promptdiffusion/README.md

Zhendong-Wang commented 7 months ago

Hi there, thanks for pointing this out. I am trying to convert the checkpoints, and will let you know when it is done.

Zhendong-Wang commented 6 months ago

Hi iczaw, thank you so much for your contribution in merging our framework into HF diffusers. I am very sorry for the delayed reply and it seems that you have already finished the conversion and merge that PR. I tried your repo and now it shows missing the model_index.py file, which is actually in your base folder. I will fix the issue, use your conversion and share it through my HF as well. I will also mention the use of the Prompt-Diffusion Pipeline in this github page and your contribution in this. Thanks again for your help here!