updating Stable Diffusion to 2.1? about arldm HOT 3 CLOSED

Hi @KyonP , sorry for the delayed reply, I was busy with ICCV last week. I don't think you should modify the unet architecture, this will cause a loading failure (shape dismatch). Instead, you can add a linear layer between the CLIP and BLIP conditon and the Unet of stable diffusion 2.1. Also you may need to modify some of the diffusion code with reference of stable diffusion 2.1 official implementation.

from arldm.

Thank you for your reply.

I thought you submitted this work at a conference and working on a process (CVPR rebuttal or ICCV submission) :)
I hope you have good news.

I hacked into your code to upgrade SD 2.1, and unfortunately, failed as you expected, too many lines to modify.

I will try your advice.

from arldm.

@KyonP Great! Thanks for your understanding! If you have further question on moving to SD 2.1, feel free to leave me a message~

from arldm.