Ryan Dick db9c0cad7c Replace custom RMSNorm implementation with torch.nn.functional.rms_norm(...) for improved speed.

2024-11-29 12:32:50 -05:00

..

autoencoder.py

Add FLUX VAE support to ImageToLatentsInvocation.

2024-09-02 09:38:17 -04:00

conditioner.py

Attribute black-forest-labs/flux for much of the flux code

2024-08-26 20:17:50 -04:00

layers.py

Replace custom RMSNorm implementation with torch.nn.functional.rms_norm(...) for improved speed.

2024-11-29 12:32:50 -05:00