code to store the K,V values inside the U-Net?

cumulo-autumn / StreamDiffusion

StreamDiffusion: A Pipeline-Level Solution for Real-Time Interactive Generation

Apache License 2.0

9.7k stars 692 forks source link

code to store the K,V values inside the U-Net? #133

Closed imyangs closed 8 months ago

imyangs commented 8 months ago

Dear author,

As you mentioned in the paper, you modified the UNet to reuse the Key and Value pairs, but based on your code, I found that you simply reused the UNet from diffusers. So, I would like to know if you haven't open-sourced this part or if diffusers already support it.

Looking forward to your feedback. Thank you.