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