Simon Lui 4a0c4ce4ef Some fixes to generalize CUDA specific functionality to Intel or other GPUs.

2023-09-02 18:22:10 -07:00

..

diffusionmodules

Some fixes to generalize CUDA specific functionality to Intel or other GPUs.

2023-09-02 18:22:10 -07:00

distributions

Initial commit.

2023-01-16 22:37:14 -05:00

encoders

Cleanup: Remove a bunch of useless files.

2023-06-13 02:19:08 -04:00

attention.py

Some fixes to generalize CUDA specific functionality to Intel or other GPUs.

2023-09-02 18:22:10 -07:00

ema.py

Initial commit.

2023-01-16 22:37:14 -05:00

sub_quadratic_attention.py

Disable autocast in unet for increased speed.

2023-07-05 21:58:29 -04:00