InvokeAI/ldm/modules/diffusionmodules
Mihai 071f65a892
Enable even larger images with one simple torch.nn.functional.silu import (#653)
Fixes:
File "stable-diffusion/ldm/modules/diffusionmodules/model.py", line 37, in nonlinearity
    return x*torch.sigmoid(x)
RuntimeError: CUDA out of memory. Tried to allocate 1.56 GiB [..]

Now up to 1536x1280 is possible on 8GB VRAM.
Also remove unused SiLU class.
2022-09-17 18:03:52 -04:00
..
__init__.py add code 2021-12-21 03:23:41 +01:00
model.py Enable even larger images with one simple torch.nn.functional.silu import (#653) 2022-09-17 18:03:52 -04:00
openaimodel.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
util.py Enable even larger images with one simple torch.nn.functional.silu import (#653) 2022-09-17 18:03:52 -04:00