diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-25 05:23:10 +0000 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-25 06:13:04 +0000 |
| commit | e3b53fd295aca784253dfc8668ec87b537a72f43 (patch) | |
| tree | 6fb26afd730c0561a2506ead2d2c8295d326de40 /modules/deepbooru_model.py | |
| parent | 84d9ce30cb427759547bc7876ed80ab91787d175 (diff) | |
| download | stable-diffusion-webui-gfx803-e3b53fd295aca784253dfc8668ec87b537a72f43.tar.gz stable-diffusion-webui-gfx803-e3b53fd295aca784253dfc8668ec87b537a72f43.tar.bz2 stable-diffusion-webui-gfx803-e3b53fd295aca784253dfc8668ec87b537a72f43.zip | |
Add UI setting for upcasting attention to float32
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.
Diffstat (limited to 'modules/deepbooru_model.py')
0 files changed, 0 insertions, 0 deletions