InvokeAI/ldm/modules
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
..
diffusionmodules Enable even larger images with one simple torch.nn.functional.silu import (#653) 2022-09-17 18:03:52 -04:00
distributions prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
encoders add support for Apple hardware using MPS acceleration 2022-08-31 00:33:23 -04:00
image_degradation prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
losses prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
attention.py Refactor attention.CrossAttention to remove duplicate code and apply optimizations 2022-09-17 20:19:21 +03:00
ema.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
embedding_manager.py respect --outdir again; fix issue #628 2022-09-16 19:58:45 -04:00
x_transformer.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00