InvokeAI/ldm/invoke/generator
Damian Stewart 786b8878d6
Save and display per-token attention maps (#1866)
* attention maps saving to /tmp

* tidy up diffusers branch backporting of cross attention refactoring

* base64-encoding the attention maps image for generationResult

* cleanup/refactor conditioning.py

* attention maps and tokens being sent to web UI

* attention maps: restrict count to actual token count and improve robustness

* add argument type hint to image_to_dataURL function

Co-authored-by: psychedelicious <4822129+psychedelicious@users.noreply.github.com>

Co-authored-by: damian <git@damianstewart.com>
Co-authored-by: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
2022-12-10 15:57:41 +01:00
..
__init__.py rename all modules from ldm.dream to ldm.invoke 2022-10-08 11:37:23 -04:00
base.py Save and display per-token attention maps (#1866) 2022-12-10 15:57:41 +01:00
embiggen.py Update embiggen.py 2022-12-08 22:34:49 -05:00
img2img.py Enable force free GPU memory in img2img 2022-12-07 19:25:21 -05:00
inpaint.py fix startup messages and a startup crash 2022-12-01 07:42:31 -05:00
omnibus.py Merge dev into main for 2.2.0 (#1642) 2022-11-30 16:12:23 -05:00
txt2img2img.py fix --hires to support inpainting model 2022-10-27 23:12:21 -04:00
txt2img.py Save and display per-token attention maps (#1866) 2022-12-10 15:57:41 +01:00