DAAM Extension

[kousw]

Hi, I have created DAAM extension for web ui.

https://github.com/kousw/stable-diffusion-webui-daam

The original implementation did not work with some resolutions and tokens that were too long, but it should work fine.

You may be able to use it to better understand the prompting.
But it's a bit of a forced implementation, so sorry if it doesn't work.

example

prompt : "A photo of a cute cat wearing sunglasses relaxing on a beach"
attention text: "cat, sunglasses"

[ClashSAN]

awesome, definitely appreciate this!

[some9000]

Daam it's good to see this implemented. Understanding that it can only analyze whatever it just created took a minute, but it's smooth sailing from there. Really useful.

What do you think about making this show the keywords on the image? How to do it can, probably, be taken from this (delightfully documented) extension: https://github.com/Extraltodeus/test_my_prompt

[kousw]

Nice idea. I just implemented that.

[some9000]

Edit: It works after pulling the latest main updates of webui itself. Thanks, this is great.

Unfortunately it is now giving the following error message (???? is censorship):

Error executing callback before_image_saved_callback for C:\Users\????\Downloads\stable-diffusion-webui\extensions\stable-diffusion-webui-daam\scripts\daam_script.py
Traceback (most recent call last):
  File "C:\Users\????\Downloads\stable-diffusion-webui\modules\script_callbacks.py", line 121, in before_image_saved_callback
    c.callback(params)
  File "C:\Users\????\Downloads\stable-diffusion-webui\extensions\stable-diffusion-webui-daam\scripts\daam_script.py", line 102, in before_image_saved
    img : Image.Image = utils.image_overlay_heat_map(params.image, heat_map, alpha=self.alpha, caption=caption)
TypeError: image_overlay_heat_map() got an unexpected keyword argument 'alpha'

[Ehplodor]

Interesting to see visually that there is a lot of attention given to the hidden side of the sunglasses. Or is it only the central part ? Mhhh

[kousw]

Maybe only the central part.

[magpie514]

Doesn't seem to work on CPU, gonna need to wait until holidays to give it use.
Still, thanks, this was sorely needed.

[daemon]

I added CPU support in the latest DAAM (v0.0.11): https://github.com/castorini/daam

[magpie514]

That's fantastic, thank you!