Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Anton Lozhkov's avatar
    Support SD2 attention slicing (#1397) · d50e3217
    Anton Lozhkov authored 
    * Support SD2 attention slicing

* Support SD2 attention slicing

* Add more copies

* Use attn_num_head_channels in blocks

* fix-copies

* Update tests

* fix imports
    d50e3217
unet_2d_blocks.py 59.3 KB