search

castorini / daam

Diffusion attentive attribution maps for interpreting Stable Diffusion.
MIT License
669 stars 61 forks source link

Add Stable Diffusion V2 support #9

Closed daemon closed 1 year ago

daemon commented 1 year ago

Add Stable Diffusion V2 support.

federicotorrielli commented 1 year ago

This is awesome, really looking forward!

EDIT: between tomorrow and the day after the diffusion team is going to update the diffusers package, I reckon you should wait until 0.9.0

daemon commented 1 year ago

Cool, that sounds about right.

federicotorrielli commented 1 year ago

Diffusers 0.9.0 is out

daemon commented 1 year ago

Yep! I'm currently swamped with extending the paper, but I'll get around to it. If you (or anyone) wants to submit a PR for it, I'd also be happy to review.

EDIT: Actually, I'll spend a bit of time tonight to see if I can do it quickly.

federicotorrielli commented 1 year ago

I can take a look at the code for sure, but I'll bet that you and the other authors could do a better job at it than me

daemon commented 1 year ago

Updated to support V2. Check out the readme for the CLI utility as well.