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 |