Anton Lozhkov d50e321745 Support SD2 attention slicing (#1397) ...

* Support SD2 attention slicing

* Support SD2 attention slicing

* Add more copies

* Use attn_num_head_channels in blocks

* fix-copies

* Update tests

* fix imports

2022-11-24 22:42:59 +01:00

..

__init__.py

Support SD2 attention slicing (#1397)

2022-11-24 22:42:59 +01:00

test_stable_diffusion.py

Support SD2 attention slicing (#1397)

2022-11-24 22:42:59 +01:00