aboutsummaryrefslogtreecommitdiff
path: root/modules/sub_quadratic_attention.py
AgeCommit message (Collapse)Author
2023-01-25Add UI setting for upcasting attention to float32brkirch
Adds "Upcast cross attention layer to float32" option in Stable Diffusion settings. This allows for generating images using SD 2.1 models without --no-half or xFormers. In order to make upcasting cross attention layer optimizations possible it is necessary to indent several sections of code in sd_hijack_optimizations.py so that a context manager can be used to disable autocast. Also, even though Stable Diffusion (and Diffusers) only upcast q and k, unfortunately my findings were that most of the cross attention layer optimizations could not function unless v is upcast also.
2023-01-09Remove fallback for Protocol import and remove Protocol import and remove ↵AUTOMATIC
instances of Protocol in code add some whitespace between functions to be in line with other code in the repo
2023-01-07Add fallback for Protocol importProGamerGov
2023-01-06Added licensebrkirch
2023-01-06Use narrow instead of dynamic_slicebrkirch
2023-01-06Add Birch-san's sub-quadratic attention implementationbrkirch